Searched refs:GpuExecutable (Results 1 – 19 of 19) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.cc | 57 GpuExecutable::GpuExecutable(GpuExecutable::Params params) in GpuExecutable() function in xla::gpu::GpuExecutable
76 GpuExecutable::~GpuExecutable() { in ~GpuExecutable()
94 Status GpuExecutable::CheckCompatibilityWithServiceExecutableRunOptions( in CheckCompatibilityWithServiceExecutableRunOptions()
128 Status GpuExecutable::ExecuteThunks( in ExecuteThunks()
256 StatusOr<const GpuExecutable::BufferAllocToDeviceMemoryMap*>
257 GpuExecutable::ResolveConstantGlobals(se::Stream* stream) { in ResolveConstantGlobals()
302 StatusOr<se::DeviceMemoryBase> GpuExecutable::BufferForAllocation( in BufferForAllocation()
304 const GpuExecutable::BufferAllocToDeviceMemoryMap* globals, in BufferForAllocation()
374 StatusOr<BufferAllocations> GpuExecutable::GenerateBufferAllocations( in GenerateBufferAllocations()
376 const GpuExecutable::BufferAllocToDeviceMemoryMap* globals, in GenerateBufferAllocations()
[all …]
|
| D | gpu_executable.h | 50 class GpuExecutable : public Executable {
90 explicit GpuExecutable(Params params);
91 ~GpuExecutable() override;
161 const GpuExecutable::BufferAllocToDeviceMemoryMap* globals,
167 const GpuExecutable::BufferAllocToDeviceMemoryMap* globals,
219 TF_DISALLOW_COPY_AND_ASSIGN(GpuExecutable);
222 StatusOr<absl::flat_hash_map<ShapeIndex, GpuExecutable::OutputInfo>>
|
| D | kernel_thunk.h | 38 class GpuExecutable; variable
60 Status Initialize(const GpuExecutable& executable,
|
| D | ir_emitter_context.h | 73 std::vector<GpuExecutable::ConstantInfo>& constants() { return constants_; } in constants()
98 std::vector<GpuExecutable::ConstantInfo> constants_;
|
| D | thunk.h | 33 class GpuExecutable; variable
98 virtual Status Initialize(const GpuExecutable& /*executable*/, in Initialize() argument
|
| D | for_thunk.h | 40 Status Initialize(const GpuExecutable& executable,
|
| D | sequential_thunk.h | 42 Status Initialize(const GpuExecutable& executable,
|
| D | sequential_thunk.cc | 31 Status SequentialThunk::Initialize(const GpuExecutable& executable, in Initialize()
|
| D | while_thunk.h | 51 Status Initialize(const GpuExecutable& executable,
|
| D | for_thunk.cc | 38 Status ForThunk::Initialize(const GpuExecutable& executable, in Initialize()
|
| D | gpu_compiler.cc | 558 std::vector<GpuExecutable::ConstantInfo>* constants) { in CompileModuleToLlvmIrImpl()
902 std::vector<GpuExecutable::ConstantInfo> constants; in RunBackend()
933 absl::flat_hash_map<ShapeIndex, GpuExecutable::OutputInfo>; in RunBackend()
949 auto* gpu_executable = new GpuExecutable( in RunBackend()
1008 absl::flat_hash_map<ShapeIndex, GpuExecutable::OutputInfo>* output_info, in GetMlirAllocationInfo()
1107 absl::flat_hash_map<ShapeIndex, GpuExecutable::OutputInfo> output_info; in CompileLmhloToExecutable()
1144 auto* gpu_executable = new GpuExecutable( in CompileLmhloToExecutable()
|
| D | conditional_thunk.h | 65 Status Initialize(const GpuExecutable& executable,
|
| D | while_thunk.cc | 46 Status WhileThunk::Initialize(const GpuExecutable& executable, in Initialize()
|
| D | conditional_thunk.cc | 57 Status ConditionalThunk::Initialize(const GpuExecutable& executable, in Initialize()
|
| D | kernel_thunk.cc | 43 Status KernelThunk::Initialize(const GpuExecutable& executable, in Initialize()
|
| D | ir_emitter.cc | 147 GpuExecutable::ConstantInfo info; in EmitConstants()
|
| D | ir_emitter_unnested.cc | 757 GpuExecutable::ConstantInfo info; in EmitConstant()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_codegen_test.cc | 53 string ptx_str(static_cast<GpuExecutable*>(executable.get())->text()); in CompileAndOptionallyVerifyPtx()
|
| D | gemm_rewrite_test.cc | 47 GpuExecutable* gpu_executable = in CheckNumberOfAllocations()
48 static_cast<GpuExecutable*>(executable.get()); in CheckNumberOfAllocations()
|