| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | memset_thunk.cc | 24 Status MemzeroThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::MemzeroThunk
33 Status Memset32BitValueThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::Memset32BitValueThunk
|
| D | copy_thunk.cc | 32 Status HostToDeviceCopyThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::HostToDeviceCopyThunk
49 Status DeviceToDeviceCopyThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::DeviceToDeviceCopyThunk
|
| D | cudnn_batchnorm_thunk.cc | 50 Status CudnnBatchNormForwardInferenceThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::CudnnBatchNormForwardInferenceThunk
89 Status CudnnBatchNormForwardTrainingThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::CudnnBatchNormForwardTrainingThunk
137 Status CudnnBatchNormBackwardThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::CudnnBatchNormBackwardThunk
|
| D | replica_id_thunk.cc | 21 Status ReplicaOrPartitionIdThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ReplicaOrPartitionIdThunk
|
| D | nccl_collective_thunk.cc | 64 Status NcclCollectiveThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::NcclCollectiveThunk
|
| D | for_thunk.cc | 44 Status ForThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ForThunk
|
| D | custom_call_thunk.cc | 40 Status CustomCallThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::CustomCallThunk
|
| D | sequential_thunk.cc | 39 Status SequentialThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::SequentialThunk
|
| D | convolution_thunk.cc | 43 Status ConvolutionThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ConvolutionThunk
|
| D | infeed_thunk.cc | 33 Status InfeedThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::InfeedThunk
|
| D | while_thunk.cc | 54 Status WhileThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::WhileThunk
|
| D | cholesky_thunk.cc | 51 Status CholeskyThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::CholeskyThunk
|
| D | outfeed_thunk.cc | 33 Status OutfeedThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::OutfeedThunk
|
| D | conditional_thunk.cc | 70 Status ConditionalThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ConditionalThunk
|
| D | triangular_solve_thunk.cc | 71 Status TriangularSolveThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::TriangularSolveThunk
|
| D | kernel_thunk.cc | 87 Status KernelThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::KernelThunk
|
| D | fft_thunk.cc | 117 Status FftThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::FftThunk
|
| D | collective_permute_thunk.cc | 230 Status CollectivePermuteThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::CollectivePermuteThunk
|
| D | gemm_thunk.cc | 58 Status GemmThunk::ExecuteOnStream(const ExecuteParams ¶ms) { in ExecuteOnStream() function in xla::gpu::GemmThunk
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | executable.cc | 81 StatusOr<ScopedShapedBuffer> Executable::ExecuteOnStream( in ExecuteOnStream() function in xla::Executable
118 StatusOr<ExecutionOutput> Executable::ExecuteOnStream( in ExecuteOnStream() function in xla::Executable
|