Home
last modified time | relevance | path

Searched refs:jitrt_executable (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_compiler.h46 JitRtExecutableProto jitrt_executable; in FromString() local
47 if (!jitrt_executable.ParseFromString(serialized)) { in FromString()
51 new JitRtAotCompilationResult(std::move(jitrt_executable))); in FromString()
71 explicit JitRtAotCompilationResult(JitRtExecutableProto jitrt_executable) in JitRtAotCompilationResult() argument
72 : jitrt_executable_(std::move(jitrt_executable)) {} in JitRtAotCompilationResult()
H A Dgpu_executable.cc573 GpuExecutable::JitRtExecutable* jitrt_executable, in ExecuteJitRt() argument
617 void* size = const_cast<int64_t*>(&jitrt_executable->buffer_size(i)); in ExecuteJitRt()
650 run_options, &jitrt_executable->debug_options(), in ExecuteJitRt()
651 &jitrt_executable->kernels_cache(), in ExecuteJitRt()
652 &jitrt_executable->gemm_configs_cache(), &jitrt_executable->collectives(), in ExecuteJitRt()
667 runtime::Executable& executable = jitrt_executable->executable(); in ExecuteJitRt()
1012 JitRtExecutable* jitrt_executable) in GpuExecutable() argument
1020 jitrt_executable_(jitrt_executable) { in GpuExecutable()
1100 JitRtExecutable * jitrt_executable, in LoadFromObjFile()
1109 std::move(output_info), jitrt_executable)); in LoadFromObjFile()
H A Dgpu_executable.h158 JitRtExecutable* jitrt_executable);
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A DBUILD2063 "jitrt_executable",