diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2016-07-28 10:01:13 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-07-28 11:17:30 -0700 |
commit | 34c820dcc6a6c2cf2df448b17b2a801fbcb16d6b (patch) | |
tree | 19e01414a8304ceffa6fc2a84c8a30032e97160a /grpc.BUILD | |
parent | 2ba4a45f54b7a9b8d2d859da0955b811b3be9d59 (diff) |
Improve memory type deduction for operators with type list arguments.
Currently, if an operator has any type list arguments, HostMemory type annotations are ignored and the memory types for all arguments (type lists or not) are deduced from the data types.
This CL makes two changes:
* If an operator has a type list argument, only deduce memory types from data types for the type lists themselves. Use the normal rules for other arguments.
* In addition, make it possible to force type list arguments to be in HostMemory, overriding the default memory types.
Change: 128717962
Diffstat (limited to 'grpc.BUILD')
0 files changed, 0 insertions, 0 deletions