Home
last modified time | relevance | path

Searched defs:ExecuteOnStream (Results 1 – 22 of 22) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dmemset_thunk.cc23 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
Dreplica_id_thunk.cc21 Status ReplicaOrPartitionIdThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ReplicaOrPartitionIdThunk
Dcopy_thunk.cc31 Status DeviceToDeviceCopyThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::DeviceToDeviceCopyThunk
Dgemm_thunk.cc39 Status GemmThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::GemmThunk
Dfor_thunk.cc42 Status ForThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ForThunk
Dcustom_call_thunk.cc41 Status CustomCallThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::CustomCallThunk
Dcublas_lt_matmul_thunk.cc44 Status CublasLtMatmulThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::CublasLtMatmulThunk
Dwhile_thunk.cc46 Status WhileThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::WhileThunk
Dsequential_thunk.cc43 Status SequentialThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::SequentialThunk
Dinfeed_thunk.cc33 Status InfeedThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::InfeedThunk
Dconditional_thunk.cc47 Status ConditionalThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ConditionalThunk
Dconvolution_thunk.cc55 Status ConvolutionThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ConvolutionThunk
Doutfeed_thunk.cc32 Status OutfeedThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::OutfeedThunk
Dkernel_thunk.cc89 Status KernelThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::KernelThunk
Dnccl_collective_thunk.cc172 Status NcclCollectiveThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::NcclCollectiveThunk
Dcholesky_thunk.cc123 Status CholeskyThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::CholeskyThunk
Dtriangular_solve_thunk.cc75 Status TriangularSolveThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::TriangularSolveThunk
Dfft_thunk.cc85 Status FftThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::FftThunk
Dbef_thunk.cc325 Status BefThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::BefThunk
Dnccl_all_reduce_thunk.cc328 Status NcclAllReduceDoneThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::NcclAllReduceDoneThunk
Dmatmul_utils.cc717 Status MatmulPlan::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::cublas_lt::MatmulPlan
/external/tensorflow/tensorflow/compiler/xla/service/
Dexecutable.cc80 StatusOr<ScopedShapedBuffer> Executable::ExecuteOnStream( in ExecuteOnStream() function in xla::Executable
117 StatusOr<ExecutionOutput> Executable::ExecuteOnStream( in ExecuteOnStream() function in xla::Executable