| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | device_util.cc | 36 void AddDevicesToOp(mlir::Operation* op, const DeviceSet* device_set) { in AddDevicesToOp()
|
| D | device_util_test.cc | 65 DeviceSet device_set; in TEST() local
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 57 void SetDeviceSet(const DeviceSet* device_set) { device_set_ = device_set; } in SetDeviceSet()
117 DeviceSet device_set; in TEST() local
|
| D | convert_graph.cc | 577 auto device_set = params.cluster->GetDeviceSet(); in GetDeviceAndAllocator() local
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | device_util_test.cc | 28 jit::DeviceSet device_set; in PickDeviceHelper() local
110 jit::DeviceSet device_set; in SimpleRoundTripTestForDeviceSet() local
|
| D | increase_dynamism_for_auto_jit_pass_test.cc | 68 std::unique_ptr<DeviceSet> device_set(new DeviceSet()); in IncreaseDynamismForAutoJit() local
|
| D | build_xla_ops_pass.cc | 348 jit::DeviceSet device_set; in InferDeviceForCluster() local
|
| D | mark_for_compilation_pass.cc | 313 const DeviceSet& device_set, in MakeNewCluster()
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | mlir_tf2xla.cc | 103 DeviceSet device_set; in ConvertGraphDefToXlaViaMlir() local
|
| /external/tensorflow/tensorflow/tools/optimization/ |
| D | optimization_pass_runner.cc | 118 DeviceSet device_set; in Run() local
|
| /external/tensorflow/tensorflow/core/grappler/clusters/ |
| D | virtual_cluster.cc | 46 VirtualCluster::VirtualCluster(const DeviceSet* device_set) in VirtualCluster()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | graph_execution_state.h | 43 const DeviceSet* device_set = nullptr; member
|
| D | optimization_registry.h | 49 const DeviceSet* device_set = nullptr; // Not owned. member
|
| D | inspecting_placer.cc | 114 const DeviceSet* device_set, in InspectingPlacer()
|
| D | partitioning_utils.cc | 26 const DeviceSet& device_set, std::unique_ptr<Graph> graph, in PartitionFunctionGraph()
|
| D | device_set_test.cc | 50 const DeviceSet& device_set() const { return devices_; } in device_set() function in tensorflow::__anon39fffd610111::DeviceSetTest
|
| D | process_function_library_runtime.h | 192 const DeviceSet* device_set() { return &device_set_; } in device_set() function
|
| D | graph_execution_state.cc | 343 Status LookupDevice(const DeviceSet& device_set, const string& tensor_name, in LookupDevice()
|
| D | colocation_graph.cc | 576 const DeviceSet* device_set, in ColocationGraph()
|
| D | process_function_library_runtime.cc | 365 const std::vector<string>& output_devices, const DeviceSet& device_set, in PinArgsAndRets()
|
| /external/tensorflow/tensorflow/compiler/xla/client/ |
| D | client_library.cc | 83 se::Platform* platform, const absl::optional<std::set<int>>& device_set) { in GetOrCreateLocalClient()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/rpc/ |
| D | grpc_server_lib.cc | 295 std::vector<string> filtered_worker_list) { in Init()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | meta_optimizer.cc | 844 FunctionLibraryDefinition* flib, const DeviceSet& device_set, in OptimizeGraph()
|
| D | function_optimizer.cc | 1348 DeviceSet device_set; // does not own devices in InlineFunctionCalls() local
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | master.cc | 372 std::unique_ptr<DeviceSet> device_set; in CreateSession() local
|