Searched refs:GpuExecutable (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_executable.cc | 50 GpuExecutable::GpuExecutable( in GpuExecutable() function in xla::gpu::GpuExecutable
66 Status GpuExecutable::ExecuteThunks( in ExecuteThunks()
187 StatusOr<const GpuExecutable::BufferAllocToDeviceMemoryMap*>
188 GpuExecutable::ResolveConstantGlobals(se::StreamExecutor* executor) { in ResolveConstantGlobals()
236 StatusOr<ScopedShapedBuffer> GpuExecutable::ExecuteOnStream( in ExecuteOnStream()
349 StatusOr<ScopedShapedBuffer> GpuExecutable::ExecuteAsyncOnStream( in ExecuteAsyncOnStream()
357 const PointsToSet& GpuExecutable::GetRootPointsToSet() const { in GetRootPointsToSet()
|
| D | gpu_executable.h | 50 class GpuExecutable : public Executable {
54 GpuExecutable(const string& ptx, const std::vector<uint8>& cubin,
148 TF_DISALLOW_COPY_AND_ASSIGN(GpuExecutable);
|
| D | kernel_thunk.h | 38 class GpuExecutable; variable
61 Status Initialize(const GpuExecutable& executable,
|
| D | thunk.h | 31 class GpuExecutable; variable
83 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 | while_thunk.h | 49 Status Initialize(const GpuExecutable& executable,
|
| D | sequential_thunk.cc | 28 Status SequentialThunk::Initialize(const GpuExecutable& executable, in Initialize()
|
| D | conditional_thunk.h | 54 Status Initialize(const GpuExecutable& executable,
|
| D | for_thunk.cc | 37 Status ForThunk::Initialize(const GpuExecutable& executable, in Initialize()
|
| D | while_thunk.cc | 42 Status WhileThunk::Initialize(const GpuExecutable& executable, in Initialize()
|
| D | conditional_thunk.cc | 46 Status ConditionalThunk::Initialize(const GpuExecutable& executable, in Initialize()
|
| D | kernel_thunk.cc | 39 Status KernelThunk::Initialize(const GpuExecutable& executable, in Initialize()
|
| D | nvptx_compiler.cc | 792 auto* gpu_executable = new GpuExecutable( in RunBackend()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/ |
| D | gpu_codegen_test.cc | 43 string ptx_str(static_cast<GpuExecutable*>(executable.get())->ptx()); in CompileAndVerifyPtx()
|