| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | memset_thunk.cc | 24 Status MemzeroThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::MemzeroThunk
33 Status Memset32BitValueThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::Memset32BitValueThunk
|
| D | copy_thunk.cc | 33 Status HostToDeviceCopyThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::HostToDeviceCopyThunk
52 Status DeviceToDeviceCopyThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::DeviceToDeviceCopyThunk
|
| D | cudnn_batchnorm_thunk.cc | 101 Status CudnnBatchNormForwardInferenceThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::CudnnBatchNormForwardInferenceThunk
163 Status CudnnBatchNormForwardTrainingThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::CudnnBatchNormForwardTrainingThunk
247 Status CudnnBatchNormBackwardThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::CudnnBatchNormBackwardThunk
|
| D | tuple_thunk.cc | 25 Status TupleThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream() function in xla::gpu::TupleThunk
|
| D | for_thunk.cc | 43 Status ForThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream() function in xla::gpu::ForThunk
|
| D | sequential_thunk.cc | 36 Status SequentialThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::SequentialThunk
|
| D | while_thunk.cc | 50 Status WhileThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream() function in xla::gpu::WhileThunk
|
| D | convolution_thunk.cc | 44 Status ConvolutionThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::ConvolutionThunk
|
| D | conditional_thunk.cc | 59 Status ConditionalThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::ConditionalThunk
|
| D | outfeed_thunk.cc | 31 Status OutfeedThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::OutfeedThunk
|
| D | cholesky_thunk.cc | 53 Status CholeskyThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::CholeskyThunk
|
| D | nccl_all_reduce_thunk.cc | 314 Status NcclAllReduceThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::NcclAllReduceThunk
334 Status NcclAllReduceThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::NcclAllReduceThunk
|
| D | kernel_thunk.cc | 74 Status KernelThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream() function in xla::gpu::KernelThunk
|
| D | triangular_solve_thunk.cc | 71 Status TriangularSolveThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::TriangularSolveThunk
|
| D | infeed_thunk.cc | 30 Status InfeedThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream() function in xla::gpu::InfeedThunk
|
| D | fft_thunk.cc | 109 Status FftThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream() function in xla::gpu::FftThunk
|
| D | gpu_executable.cc | 236 StatusOr<ScopedShapedBuffer> GpuExecutable::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::GpuExecutable
|
| D | gemm_thunk.cc | 311 Status GemmThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream() function in xla::gpu::GemmThunk
|
| /external/tensorflow/tensorflow/compiler/xla/service/interpreter/ |
| D | executable.cc | 48 StatusOr<ScopedShapedBuffer> InterpreterExecutable::ExecuteOnStream( in ExecuteOnStream() function in xla::interpreter::InterpreterExecutable
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | executable.h | 102 virtual StatusOr<ExecutionOutput> ExecuteOnStream( in ExecuteOnStream() function
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_executable.cc | 267 StatusOr<ScopedShapedBuffer> CpuExecutable::ExecuteOnStream( in ExecuteOnStream() function in xla::cpu::CpuExecutable
|