Searched defs:composite_devices (Results 1 – 11 of 11) 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 | 249 const absl::flat_hash_map<string, const std::vector<string>*>& in ReplicatePerReplicaNodesInFunctionGraph()
|
| D | process_function_library_runtime.cc | 900 std::vector<CompositeDevice*> composite_devices; in InstantiateMultiDevice() local
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | kernel_and_device.h | 273 absl::flat_hash_map<string, const std::vector<string>*> composite_devices, in KernelAndDeviceFunc()
|
| D | execute.cc | 960 absl::flat_hash_map<string, const std::vector<string>*>& composite_devices, in ExtractFunctionInputInfo()
1077 absl::flat_hash_map<string, const std::vector<string>*> composite_devices; in GetOrCreateKernelAndDevice() local
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function.h | 694 absl::flat_hash_map<string, const std::vector<string>*> composite_devices; member
|
| D | function_test.cc | 1645 absl::flat_hash_map<string, std::vector<string>> composite_devices; in TEST() local
|
| D | function.cc | 1039 absl::flat_hash_map<string, std::vector<string>>* composite_devices) { in GetFunctionResourceInputDevice()
|
| /external/tensorflow/tensorflow/distribute/experimental/rpc/kernels/ |
| D | rpc_ops.cc | 672 absl::flat_hash_map<string, std::vector<string>> composite_devices; in Compute() local
|
| /external/tensorflow/tensorflow/core/data/ |
| D | captured_function.cc | 565 absl::flat_hash_map<string, std::vector<string>> composite_devices; in Instantiate() local
|