Searched refs:entry_function_name (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_executable.cc | 57 const std::string& entry_function_name, in CpuExecutable() argument
64 module_name_(entry_function_name) { in CpuExecutable()
76 jit_->FindCompiledSymbol(entry_function_name); in CpuExecutable()
79 CHECK(*sym) << "Symbol " << entry_function_name << " not found."; in CpuExecutable()
|
| H A D | cpu_executable.h | 51 const std::string& entry_function_name,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | gpu_compiler.cc | 879 mlir::ModuleOp mlir_module, llvm::StringRef entry_function_name, in LowerToJitRt() argument
889 mlir_module.lookupSymbol<mlir::func::FuncOp>(entry_function_name); in LowerToJitRt()
907 mlir_module, {entry_function_name.data(), entry_function_name.size()}, in LowerToJitRt()
917 entry_function_name.str(), os.str(), buffer_sizes.vec(), in LowerToJitRt()
1613 absl::string_view entry_function_name, se::StreamExecutor* stream_exec, in CompileLmhloToExecutable() argument
1618 entry_function_name.data(), entry_function_name.size()))); in CompileLmhloToExecutable()
|
| H A D | gpu_compiler.h | 192 absl::string_view entry_function_name, se::StreamExecutor* stream_exec,
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/lmhlo_to_gpu/ |
| H A D | pass_utils.h | 27 mlir::StringRef entry_function_name,
|
| H A D | pass_utils.cc | 27 mlir::StringRef entry_function_name, in ConvertLmhloToJitRt() argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/ |
| H A D | tf_tfl_passes.cc | 98 void AddConvertHloToTfPass(std::string entry_function_name, in AddConvertHloToTfPass() argument
119 mlir::mhlo::createExpandHloTuplesPass(entry_function_name)); in AddConvertHloToTfPass()
|