Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcopy_thunk.cc32 tensorflow::Status HostToDeviceCopyThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::HostToDeviceCopyThunk
50 tensorflow::Status DeviceToDeviceCopyThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::DeviceToDeviceCopyThunk
Dtuple_thunk.cc25 tensorflow::Status TupleThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::TupleThunk
Dcudnn_batchnorm_thunk.cc102 Status CudnnBatchNormForwardInferenceThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::CudnnBatchNormForwardInferenceThunk
161 Status CudnnBatchNormForwardTrainingThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::CudnnBatchNormForwardTrainingThunk
243 Status CudnnBatchNormBackwardThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::CudnnBatchNormBackwardThunk
Dfor_thunk.cc38 tensorflow::Status ForThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::ForThunk
Dsequential_thunk.cc35 tensorflow::Status SequentialThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::SequentialThunk
Dwhile_thunk.cc43 Status WhileThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream() function in xla::gpu::WhileThunk
Dconditional_thunk.cc44 Status ConditionalThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::ConditionalThunk
Dinfeed_thunk.cc33 Status InfeedThunk::ExecuteOnStream(const BufferAllocations& buffer_allocations, in ExecuteOnStream() function in xla::gpu::InfeedThunk
Dkernel_thunk.cc65 tensorflow::Status KernelThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::KernelThunk
Dconvolution_thunk.cc64 Status ConvolutionThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::ConvolutionThunk
Dfft_thunk.cc126 tensorflow::Status FftThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::FftThunk
Dgpu_executable.cc250 StatusOr<std::unique_ptr<ShapedBuffer>> GpuExecutable::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::GpuExecutable
Dgemm_thunk.cc226 tensorflow::Status GemmThunk::ExecuteOnStream( in ExecuteOnStream() function in xla::gpu::GemmThunk
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutable.cc50 StatusOr<std::unique_ptr<ShapedBuffer>> InterpreterExecutable::ExecuteOnStream( in ExecuteOnStream() function in xla::interpreter::InterpreterExecutable
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_executable.cc287 StatusOr<std::unique_ptr<ShapedBuffer>> CpuExecutable::ExecuteOnStream( in ExecuteOnStream() function in xla::cpu::CpuExecutable
Dparallel_cpu_executable.cc452 StatusOr<std::unique_ptr<ShapedBuffer>> ParallelCpuExecutable::ExecuteOnStream( in ExecuteOnStream() function in xla::cpu::ParallelCpuExecutable