| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | replicate_per_replica_nodes_test.cc | 91 composite_devices = {{"TPU_COMPOSITE:0", &underlying_devices}}; in TEST() local
109 ReplicatePerReplicaNodesInFunctionGraph(composite_devices, &graph)); in TEST()
136 composite_devices = {{"TPU_COMPOSITE:0", &underlying_devices}}; in TEST() local
150 ReplicatePerReplicaNodesInFunctionGraph(composite_devices, &graph)); in TEST()
177 composite_devices = {{"TPU_COMPOSITE:0", &underlying_devices_0}, in TEST() local
201 ReplicatePerReplicaNodesInFunctionGraph(composite_devices, &graph)); in TEST()
223 composite_devices = {{"TPU_COMPOSITE:0", &underlying_devices}}; in TEST() local
268 ReplicatePerReplicaNodesInFunctionGraph(composite_devices, &graph)); in TEST()
301 composite_devices = {{"TPU_COMPOSITE:0", &underlying_devices}}; in TEST() local
315 ReplicatePerReplicaNodesInFunctionGraph(composite_devices, &graph)); in TEST()
|
| D | replicate_per_replica_nodes.cc | 250 composite_devices, in ReplicatePerReplicaNodesInFunctionGraph()
253 for (const auto& it : composite_devices) { in ReplicatePerReplicaNodesInFunctionGraph()
273 *composite_devices.at(it.first); in ReplicatePerReplicaNodesInFunctionGraph()
|
| D | replicate_per_replica_nodes.h | 39 composite_devices,
|
| D | optimization_registry.h | 54 const std::vector<CompositeDevice*>* composite_devices = member
|
| D | process_function_library_runtime.cc | 900 std::vector<CompositeDevice*> composite_devices; in InstantiateMultiDevice() local
903 for (auto* d : composite_devices_) composite_devices.push_back(d); in InstantiateMultiDevice()
905 optimization_options.composite_devices = &composite_devices; in InstantiateMultiDevice()
959 options.composite_devices, graph.get())); in InstantiateMultiDevice()
|
| D | process_function_library_runtime_test.cc | 919 inst_opts.composite_devices[composite_device->name()] = in TEST_F()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | captured_function.cc | 565 absl::flat_hash_map<string, std::vector<string>> composite_devices; in Instantiate() local
603 if (composite_devices.find(composite_device) == in Instantiate()
604 composite_devices.end()) { in Instantiate()
606 composite_devices[composite_device].push_back( in Instantiate()
629 for (const auto& it : composite_devices) { in Instantiate()
630 inst_opts.composite_devices[it.first] = &it.second; in Instantiate()
|
| /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() argument
299 composite_devices_(std::move(composite_devices)), in KernelAndDeviceFunc()
|
| D | execute.cc | 960 absl::flat_hash_map<string, const std::vector<string>*>& composite_devices, in ExtractFunctionInputInfo() argument
988 composite_devices[input_device->name()] = in ExtractFunctionInputInfo()
1077 absl::flat_hash_map<string, const std::vector<string>*> composite_devices; in GetOrCreateKernelAndDevice() local
1087 op, kernel_def, input_device_ptrs, composite_devices, in GetOrCreateKernelAndDevice()
1225 std::move(composite_devices), in GetOrCreateKernelAndDevice()
|
| D | kernel_and_device.cc | 169 options.composite_devices = composite_devices_; in InstantiateFunc()
|
| /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
676 captured[i], num_non_captured_inputs + i, *fdef, &composite_devices)); in Compute()
682 for (const auto& it : composite_devices) { in Compute()
683 instantiate_opts.composite_devices[it.first] = &it.second; in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | function_test.cc | 1645 absl::flat_hash_map<string, std::vector<string>> composite_devices; in TEST() local
1659 arg0, /*arg_index=*/0, fdef, &composite_devices); in TEST()
1661 arg1, /*arg_index=*/1, fdef, &composite_devices); in TEST()
1665 EXPECT_EQ(composite_devices.size(), 1); in TEST()
1666 EXPECT_EQ(composite_devices.at("/device:COMPOSITE:0").size(), 2); in TEST()
|
| D | function.h | 694 absl::flat_hash_map<string, const std::vector<string>*> composite_devices; member
978 absl::flat_hash_map<string, std::vector<string>>* composite_devices);
|
| D | function.cc | 1039 absl::flat_hash_map<string, std::vector<string>>* composite_devices) { in GetFunctionResourceInputDevice() argument
1051 if (composite_devices->find(composite_device) == composite_devices->end()) { in GetFunctionResourceInputDevice()
1053 (*composite_devices)[composite_device].push_back(handles(i).device()); in GetFunctionResourceInputDevice()
|