| /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 | 33 Status HostToDeviceCopyThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::HostToDeviceCopyThunk
51 Status DeviceToDeviceCopyThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::DeviceToDeviceCopyThunk
|
| D | replica_id_thunk.cc | 25 Status ReplicaIdThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ReplicaIdThunk
|
| D | tuple_thunk.cc | 25 Status TupleThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::TupleThunk
|
| D | cudnn_batchnorm_thunk.cc | 117 Status CudnnBatchNormForwardInferenceThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::CudnnBatchNormForwardInferenceThunk
167 Status CudnnBatchNormForwardTrainingThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::CudnnBatchNormForwardTrainingThunk
235 Status CudnnBatchNormBackwardThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::CudnnBatchNormBackwardThunk
|
| D | sequential_thunk.cc | 36 Status SequentialThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::SequentialThunk
|
| D | for_thunk.cc | 43 Status ForThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ForThunk
|
| D | dummy_all_reduce_thunk.cc | 29 Status NcclAllReduceThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::NcclAllReduceThunk
|
| D | while_thunk.cc | 50 Status WhileThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::WhileThunk
|
| D | convolution_thunk.cc | 44 Status ConvolutionThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ConvolutionThunk
|
| D | conditional_thunk.cc | 59 Status ConditionalThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ConditionalThunk
|
| D | cholesky_thunk.cc | 53 Status CholeskyThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::CholeskyThunk
|
| D | outfeed_thunk.cc | 31 Status OutfeedThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::OutfeedThunk
|
| D | kernel_thunk.cc | 72 Status KernelThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::KernelThunk
|
| D | triangular_solve_thunk.cc | 71 Status TriangularSolveThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::TriangularSolveThunk
|
| D | infeed_thunk.cc | 30 Status InfeedThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::InfeedThunk
|
| D | custom_call_thunk.cc | 124 Status CustomCallThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::CustomCallThunk
|
| D | fft_thunk.cc | 109 Status FftThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::FftThunk
|
| D | collective_permute_thunk.cc | 228 Status CollectivePermuteThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::CollectivePermuteThunk
|
| D | gemm_thunk.cc | 49 Status GemmThunk::ExecuteOnStream(const ExecuteParams ¶ms) { in ExecuteOnStream() function in xla::gpu::GemmThunk
|
| D | nccl_all_reduce_thunk.cc | 487 Status NcclAllReduceThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::NcclAllReduceThunk
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | executable.cc | 35 StatusOr<ScopedShapedBuffer> Executable::ExecuteOnStream( in ExecuteOnStream() function in xla::Executable
74 StatusOr<ExecutionOutput> Executable::ExecuteOnStream( in ExecuteOnStream() function in xla::Executable
|