| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| H A D | memset_thunk.cc | 23 Status MemzeroThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::MemzeroThunk
30 Status Memset32BitValueThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::Memset32BitValueThunk
|
| H A D | replica_id_thunk.cc | 21 Status ReplicaOrPartitionIdThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ReplicaOrPartitionIdThunk
|
| H A D | gemm_thunk.cc | 39 Status GemmThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::GemmThunk
|
| H A D | copy_thunk.cc | 31 Status DeviceToDeviceCopyThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::DeviceToDeviceCopyThunk
|
| H A D | for_thunk.cc | 42 Status ForThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ForThunk
|
| H A D | custom_call_thunk.cc | 41 Status CustomCallThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::CustomCallThunk
|
| H A D | cublas_lt_matmul_thunk.cc | 44 Status CublasLtMatmulThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::CublasLtMatmulThunk
|
| H A D | while_thunk.cc | 46 Status WhileThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::WhileThunk
|
| H A D | sequential_thunk.cc | 43 Status SequentialThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::SequentialThunk
|
| H A D | infeed_thunk.cc | 33 Status InfeedThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::InfeedThunk
|
| H A D | conditional_thunk.cc | 47 Status ConditionalThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ConditionalThunk
|
| H A D | convolution_thunk.cc | 55 Status ConvolutionThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ConvolutionThunk
|
| H A D | outfeed_thunk.cc | 32 Status OutfeedThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::OutfeedThunk
|
| H A D | kernel_thunk.cc | 89 Status KernelThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::KernelThunk
|
| H A D | nccl_collective_thunk.cc | 172 Status NcclCollectiveThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::NcclCollectiveThunk
|
| H A D | cholesky_thunk.cc | 123 Status CholeskyThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::CholeskyThunk
|
| H A D | triangular_solve_thunk.cc | 75 Status TriangularSolveThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::TriangularSolveThunk
|
| H A D | fft_thunk.cc | 85 Status FftThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::FftThunk
|
| H A D | bef_thunk.cc | 325 Status BefThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::BefThunk
|
| H A D | nccl_all_reduce_thunk.cc | 328 Status NcclAllReduceDoneThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::NcclAllReduceDoneThunk
|
| H A D | matmul_utils.cc | 717 Status MatmulPlan::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::cublas_lt::MatmulPlan
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | executable.cc | 80 StatusOr<ScopedShapedBuffer> Executable::ExecuteOnStream( in ExecuteOnStream() function in xla::Executable
117 StatusOr<ExecutionOutput> Executable::ExecuteOnStream( in ExecuteOnStream() function in xla::Executable
|