| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | mlir_bridge_pass.cc | 67 bool HasTPUDevice(const DeviceSet& device_set) { in HasTPUDevice()
87 const DeviceSet* device_set, const ConfigProto& config_proto, in GetPassState()
141 bool MlirBridgeV1CompatPass::IsEnabled(const DeviceSet* device_set, in IsEnabled()
|
| D | mlir_tf2xla.cc | 151 DeviceSet device_set; in ConvertGraphDefToXlaViaMlir() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | distributed_tpu_rewrite_helpers.cc | 30 const string& system_spec_string, const DeviceSet& device_set, in GetSystemDevice()
97 const DeviceNameUtils::ParsedName& system_spec, const DeviceSet& device_set, in GetHostSystemDevices()
151 const DeviceNameUtils::ParsedName& system_spec, const DeviceSet& device_set, in GetTPUDevices()
201 const string& node_type, Graph* graph, const DeviceSet& device_set, in ForConfigurationNodeMatchingType()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | function_optimization_registry_pass_failure_test.cc | 31 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
47 DeviceSet device_set; in TEST() local
|
| D | function_optimization_registration_test.cc | 30 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
47 DeviceSet device_set; in TEST() local
|
| D | function_optimization_registry_test.cc | 32 Status Run(const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
48 DeviceSet device_set; in TEST() local
|
| D | function_optimization_registry_no_pass_test.cc | 29 DeviceSet device_set; in TEST() local
|
| D | function_optimization_registry.cc | 27 const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
|
| 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 | 115 const DeviceSet* device_set, in InspectingPlacer()
|
| D | device_set_test.cc | 50 const DeviceSet& device_set() const { return devices_; } in device_set() function in tensorflow::__anon7d5d833d0111::DeviceSetTest
|
| D | partitioning_utils.cc | 29 const DeviceSet& device_set, std::unique_ptr<Graph> graph, in PartitionFunctionGraph()
|
| D | process_function_library_runtime.h | 205 const std::shared_ptr<DeviceSet> device_set() const { in device_set() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | graph_optimization_pass.h | 30 ::tensorflow::MlirOptimizationPassState GetPassState( in GetPassState()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| D | graph_decompose_pass.cc | 34 const DeviceSet* device_set, const ConfigProto& config_proto, in GetPassState()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_graph_test.cc | 58 void SetDeviceSet(const DeviceSet* device_set) { device_set_ = device_set; } in SetDeviceSet()
118 DeviceSet device_set; in TEST() 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
|
| /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/compiler/mlir/tensorflow/utils/ |
| D | device_util.cc | 118 void AddDevicesToOp(mlir::Operation* op, const DeviceSet* device_set) { in AddDevicesToOp()
|
| D | device_util_test.cc | 75 DeviceSet device_set; in TEST() local
|
| /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/compiler/mlir/ |
| D | mlir_graph_optimization_pass.cc | 120 const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_distributed_test.cc | 433 tensorflow::Status Run(const tensorflow::DeviceSet& device_set, in Run()
|