| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/ |
| H A D | tpu_global_init.cc | 71 DeviceSet* device_set) { in DeviceSetFromDeviceMgr()
91 const DeviceSet& device_set, in ConstructDistributedInitializationGraph()
139 const DeviceSet& device_set, Env* env, in InitializeTPUSystemGlobally()
205 DeviceSet device_set; in InitializeTPUSystemGlobally() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A 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()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | mlir_bridge_pass.cc | 92 bool HasTPUDevice(const DeviceSet& device_set) { in HasTPUDevice()
167 const DeviceSet* device_set, const ConfigProto& config_proto, in GetPassState()
266 const DeviceSet* device_set, const ConfigProto& config_proto, in GetPassState()
|
| H A D | mlir_tf2xla.cc | 151 DeviceSet device_set; in ConvertGraphDefToXlaViaMlir() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | partitioning_utils.cc | 38 const DeviceSet& device_set, Graph* graph, in PartitionFunctionGraph()
76 const DeviceSet& device_set, std::unique_ptr<Graph> graph, in PartitionFunctionGraph()
102 const DeviceSet& device_set, std::unique_ptr<Graph> graph) { in InsertTransferOps()
|
| H A 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
|
| H A 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
|
| H A 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
|
| H A D | function_optimization_registry_no_pass_test.cc | 29 DeviceSet device_set; in TEST() local
|
| H A D | function_optimization_registry.cc | 29 const DeviceSet& device_set, const ConfigProto& config_proto, in Run()
|
| H A D | graph_execution_state.h | 43 const DeviceSet* device_set = nullptr; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_graph_test.cc | 48 void SetDeviceSet(const DeviceSet* device_set) { device_set_ = device_set; } in SetDeviceSet()
107 DeviceSet device_set; in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | device_util_test.cc | 28 jit::DeviceSet device_set; in PickDeviceHelper() local
110 jit::DeviceSet device_set; in SimpleRoundTripTestForDeviceSet() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_replication_analysis.cc | 538 for (const absl::Span<const int64_t>& device_set : device_sets) { in PartiallyReplicated() local
545 for (const absl::Span<const int64_t>& device_set : device_sets) { in PartiallyReplicated() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/cuda/ |
| H A D | device_set.h | 9 using device_set = std::bitset<C10_COMPILE_TIME_MAX_GPUS>; variable
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| H A D | graph_optimization_pass.h | 30 ::tensorflow::MlirOptimizationPassState GetPassState( in GetPassState()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/fallback/ |
| H A D | fallback_state.h | 54 const DeviceSet &device_set() const { return device_set_; } in device_set() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/integration/ |
| H A D | graph_decompose_pass.cc | 34 const DeviceSet* device_set, const ConfigProto& config_proto, in GetPassState()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_graph_to_mlir_pass.cc | 68 const DeviceSet& device_set, bool is_func, in RunOnGraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/optimization/ |
| H A D | optimization_pass_runner.cc | 118 DeviceSet device_set; in Run() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/clusters/ |
| H A D | virtual_cluster.cc | 46 VirtualCluster::VirtualCluster(const DeviceSet* device_set) in VirtualCluster()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/kernels/ |
| H A D | tpu_functional_ops.cc | 2396 const DeviceSet& device_set, in PlacementHelper()
2412 const DeviceSet& device_set, in PartitionHelper()
2480 Status TPUPartitionedCallOp::SetDeviceOrdinal(const DeviceSet& device_set, in SetDeviceOrdinal()
2554 const DeviceSet& device_set, int replica_id, uint64 cache_hash, in InstantiateFunctionsFromSubgraphs()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/eager/ |
| H A D | function_cache.cc | 47 const tensorflow::DeviceSet& device_set, in GetOrAddFunction()
|
| H A D | transform_graph_function.cc | 37 const tensorflow::DeviceSet& device_set, in TransformGraphFunction()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| H A D | device_util.cc | 115 void AddDevicesToOp(mlir::Operation* op, const DeviceSet* device_set) { in AddDevicesToOp()
|