Searched defs:composite_devices (Results 1 – 10 of 10) 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
223 composite_devices = {{"TPU_COMPOSITE:0", &underlying_devices}}; in TEST() local
301 composite_devices = {{"TPU_COMPOSITE:0", &underlying_devices}}; in TEST() local
|
| D | optimization_registry.h | 54 const std::vector<CompositeDevice*>* composite_devices = member
|
| D | replicate_per_replica_nodes.cc | 253 const absl::flat_hash_map<string, const std::vector<string>*>& in ReplicatePerReplicaNodesInFunctionGraph()
|
| D | process_function_library_runtime.cc | 803 std::vector<CompositeDevice*> composite_devices; in InstantiateMultiDevice() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 268 absl::flat_hash_map<string, const std::vector<string>*> composite_devices, in KernelAndDeviceFunc()
|
| D | execute.cc | 814 absl::flat_hash_map<string, const std::vector<string>*> composite_devices; in GetOrCreateKernelAndDevice() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 692 absl::flat_hash_map<string, const std::vector<string>*> composite_devices; member
|
| D | function_test.cc | 1582 absl::flat_hash_map<string, std::vector<string>> composite_devices; in TEST() local
|
| D | function.cc | 1000 absl::flat_hash_map<string, std::vector<string>>* composite_devices) { in GetFunctionResourceInputDevice()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | captured_function.cc | 546 absl::flat_hash_map<string, std::vector<string>> composite_devices; in Instantiate() local
|