Searched refs:cpu_executable (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | xla_jit_compiled_cpu_function.cc | 124 const xla::cpu::CpuExecutable* cpu_executable = in Compile() local
127 cpu_executable->compute_function(); in Compile()
129 cpu_executable->buffer_assignment(); in Compile()
184 if (cpu_executable->hlo_profiling_enabled()) { in Compile()
186 &jit->static_data_, &cpu_executable->hlo_profile_printer_data()); in Compile()
189 cpu_executable->hlo_profile_printer_data().profile_counters_size()); in Compile()
|
| H A D | BUILD | 384 "//tensorflow/compiler/xla/service/cpu:cpu_executable",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | cpu_compiler.cc | 1271 auto cpu_executable = std::make_unique<CpuExecutable>( in CompileLegacyCpuExecutable() local
1276 cpu_executable->set_ir_module_string(ir_module_string); in CompileLegacyCpuExecutable()
1282 DumpingEnabledForHloModule(cpu_executable->module())) { in CompileLegacyCpuExecutable()
1284 *hlo_proto->mutable_hlo_module() = cpu_executable->module().ToProto(); in CompileLegacyCpuExecutable()
1286 cpu_executable->buffer_assignment().ToProto(); in CompileLegacyCpuExecutable()
1287 cpu_executable->set_hlo_proto(std::move(hlo_proto)); in CompileLegacyCpuExecutable()
1290 return cpu_executable; in CompileLegacyCpuExecutable()
1307 std::unique_ptr<CpuExecutable> cpu_executable; in RunBackend() local
1308 TF_ASSIGN_OR_RETURN(cpu_executable, in RunBackend()
1311 cpu_executable->set_debug_info( in RunBackend()
[all …]
|
| H A D | BUILD | 182 ":cpu_executable",
456 name = "cpu_executable",
457 srcs = ["cpu_executable.cc"],
458 hdrs = ["cpu_executable.h"],
1316 ":cpu_executable",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| H A D | tfrt_cpu_pjrt_client.cc | 389 TF_ASSIGN_OR_RETURN(std::unique_ptr<Executable> cpu_executable, in Compile()
393 tensorflow::down_cast<cpu::CpuExecutable*>(cpu_executable.get()); in Compile()
407 cpu_executable->module())); in Compile()
411 options.parameter_is_tupled_arguments, std::move(cpu_executable), in Compile()
1161 std::unique_ptr<Executable> cpu_executable, in TfrtCpuExecutable() argument
1171 cpu_executable_(std::move(cpu_executable)), in TfrtCpuExecutable()
1431 auto* cpu_executable = in ExecuteHelper() local
1435 CreateBufferTable(cpu_executable->buffer_assignment(), tracked_buffers)); in ExecuteHelper()
1489 cpu_executable->compute_function()(result_buffer, &run_options, nullptr, in ExecuteHelper()
1518 [cpu_executable, result_buffer, in ExecuteHelper()
[all …]
|
| H A D | tfrt_cpu_pjrt_client.h | 474 std::unique_ptr<Executable> cpu_executable,
|
| H A D | BUILD | 585 "//tensorflow/compiler/xla/service/cpu:cpu_executable",
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | BUILD | 806 "//tensorflow/compiler/xla/service/cpu:cpu_executable",
|