Searched defs:composite_devices (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | replicate_per_replica_nodes_test.cc | 91 composite_devices = {{"TPU_COMPOSITE:0", &underlying_devices}}; in TEST() local
136 composite_devices = {{"TPU_COMPOSITE:0", &underlying_devices}}; in TEST() local
177 composite_devices = {{"TPU_COMPOSITE:0", &underlying_devices_0}, in TEST() local
227 composite_devices = {{"TPU_COMPOSITE:0", &underlying_devices}}; in TEST() local
305 composite_devices = {{"TPU_COMPOSITE:0", &underlying_devices}}; in TEST() local
|
| D | replicate_per_replica_nodes.cc | 221 const absl::flat_hash_map<string, const std::vector<string>*>& in ReplicatePerReplicaNodesInFunctionGraph()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 263 absl::flat_hash_map<string, const std::vector<string>*> composite_devices, in KernelAndDeviceFunc()
|
| D | execute.cc | 399 absl::flat_hash_map<string, const std::vector<string>*> composite_devices; in GetOrCreateKernelAndDevice() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 672 absl::flat_hash_map<string, const std::vector<string>*> composite_devices; member
|
| D | function_test.cc | 1578 absl::flat_hash_map<string, std::vector<string>> composite_devices; in TEST() local
|
| D | function.cc | 998 absl::flat_hash_map<string, std::vector<string>>* composite_devices) { in GetFunctionResourceInputDevice()
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | captured_function.cc | 609 absl::flat_hash_map<string, std::vector<string>> composite_devices; in Instantiate() local
|