Home
last modified time | relevance | path

Searched refs:device_specs (Results 1 – 13 of 13) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/dtensor/python/
H A Dmesh_util.py94 device_specs, device_type = _make_device_specs(devices, device_type)
97 device_specs = [local_spec.make_merged_spec(d) for d in device_specs]
100 mesh_dims = [('x', len(device_specs))]
103 mesh_dims[0] = (mesh_dims[0][0], len(device_specs))
108 if np.prod(shape) != len(device_specs):
112 global_device_ids = np.arange(len(device_specs)).reshape(shape)
118 local_devices=device_specs,
121 num_global_devices=len(device_specs),
169 device_specs, device_type = _make_device_specs(local_devices, device_type)
182 device_specs = [local_spec.make_merged_spec(d) for d in device_specs]
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Dtarget_annotation.cc53 explicit TargetAnnotationPass(llvm::ArrayRef<std::string> device_specs) in TargetAnnotationPass() argument
55 device_specs_flag_ = device_specs; in TargetAnnotationPass()
64 llvm::ArrayRef<std::string> device_specs,
83 Operation* op, llvm::ArrayRef<std::string> device_specs, in SetTargetAnnotation() argument
92 for (const auto& device : device_specs) { in SetTargetAnnotation()
141 llvm::ArrayRef<std::string> device_specs) { in CreateTargetAnnotationPass() argument
142 return std::make_unique<TargetAnnotationPass>(device_specs); in CreateTargetAnnotationPass()
H A Dget_alternative_subgraph.cc108 explicit AlternativeSubgraphPass(llvm::ArrayRef<std::string> device_specs) { in AlternativeSubgraphPass() argument
109 device_specs_flag_ = device_specs; in AlternativeSubgraphPass()
281 std::vector<std::string> device_specs; in runOnOperation() local
282 if (!ProcessTargetDevices(device_specs_flag_, &device_specs)) { in runOnOperation()
299 GetAlternativeGraphForFunc(device_specs, func, module, &builder); in runOnOperation()
306 llvm::ArrayRef<std::string> device_specs) { in CreateAlternativeSubgraphPass() argument
307 return std::make_unique<AlternativeSubgraphPass>(device_specs); in CreateAlternativeSubgraphPass()
H A Dpasses.h34 llvm::ArrayRef<std::string> device_specs);
45 llvm::ArrayRef<std::string> device_specs);
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/hardwares/
H A Dtarget_hardware.cc269 std::vector<std::string>* device_specs) { in ProcessTargetDevices() argument
275 device_specs->push_back(device); in ProcessTargetDevices()
278 device_specs->push_back("CPU"); in ProcessTargetDevices()
282 for (const std::string& device : *device_specs) { in ProcessTargetDevices()
H A Dtarget_hardware.h189 std::vector<std::string>* device_specs);
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/py_wrapper/
H A Dtac_wrapper_pybind11.cc28 const std::vector<std::string>& device_specs, in PYBIND11_MODULE()
30 return ::tflite::run_tac(model_file_path, device_specs, in PYBIND11_MODULE()
H A Dtac_wrapper.cc55 const std::vector<std::string>& device_specs, in run_tac() argument
58 options.hardware_backends = device_specs; in run_tac()
H A Dtac_wrapper.h37 const std::vector<std::string>& device_specs,
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/
H A Dtflite_import_export.cc36 llvm::ArrayRef<std::string> device_specs) { in AttachCostPerDevice() argument
38 for (const auto& device_spec : device_specs) { in AttachCostPerDevice()
H A Dtac_translate.cc90 opt<std::string> device_specs( variable
122 absl::StrSplit(device_specs, ',', absl::SkipEmpty()); in TargetAwareConversionMain()
H A Dtac_module.cc44 llvm::ArrayRef<std::string> device_specs) { in AddTACPass() argument
50 mlir::TFL::tac::CreateAlternativeSubgraphPass(device_specs)); in AddTACPass()
H A Dtac_module.h96 llvm::ArrayRef<std::string> device_specs);