Home
last modified time | relevance | path

Searched defs:device_specs (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
Dtarget_annotation.cc48 explicit TargetAnnotationPass(llvm::ArrayRef<std::string> device_specs) in TargetAnnotationPass()
78 Operation* op, llvm::ArrayRef<std::string> device_specs, in SetTargetAnnotation()
133 llvm::ArrayRef<std::string> device_specs) { in CreateTargetAnnotationPass()
Dget_alternative_subgraph.cc96 explicit AlternativeSubgraphPass(llvm::ArrayRef<std::string> device_specs) { in AlternativeSubgraphPass()
263 std::vector<std::string> device_specs; in runOnOperation() local
288 llvm::ArrayRef<std::string> device_specs) { in CreateAlternativeSubgraphPass()
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/py_wrapper/
Dtac_wrapper_pybind11.cc27 const std::string& model_output_path) { in PYBIND11_MODULE()
Dtac_wrapper.cc54 const std::vector<std::string>& device_specs, in run_tac()
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/
Dtac_translate.cc90 opt<std::string> device_specs( variable
Dtflite_import_export.cc36 llvm::ArrayRef<std::string> device_specs) { in AttachCostPerDevice()
Dtac_module.cc42 llvm::ArrayRef<std::string> device_specs) { in AddTACPass()
/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/hardwares/
Dtarget_hardware.cc266 std::vector<std::string>* device_specs) { in ProcessTargetDevices()