| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | while_thunk.cc | 54 Status WhileThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::WhileThunk
67 TF_RETURN_IF_ERROR(condition_thunk_sequence_->ExecuteOnStream(params)); in ExecuteOnStream()
91 TF_RETURN_IF_ERROR(body_thunk_sequence_->ExecuteOnStream(params)); in ExecuteOnStream()
|
| 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 | cudnn_batchnorm_thunk.h | 64 Status ExecuteOnStream(const ExecuteParams& params) override;
92 Status ExecuteOnStream(const ExecuteParams& params) override;
120 Status ExecuteOnStream(const ExecuteParams& params) override;
|
| D | sequential_thunk.cc | 39 Status SequentialThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::SequentialThunk
44 TF_RETURN_IF_ERROR(thunk->ExecuteOnStream(params)); in ExecuteOnStream()
|
| D | memset_thunk.h | 39 Status ExecuteOnStream(const ExecuteParams& params) override;
55 Status ExecuteOnStream(const ExecuteParams& params) override;
|
| D | copy_thunk.h | 43 Status ExecuteOnStream(const ExecuteParams& params) override;
65 Status ExecuteOnStream(const ExecuteParams& params) override;
|
| D | for_thunk.cc | 44 Status ForThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ForThunk
51 TF_RETURN_IF_ERROR(body_thunk_sequence_->ExecuteOnStream(params)); in ExecuteOnStream()
|
| 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 | conditional_thunk.cc | 70 Status ConditionalThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ConditionalThunk
104 config_.branch_thunks[branch_index]->ExecuteOnStream(params)); in ExecuteOnStream()
|
| D | infeed_thunk.h | 42 Status ExecuteOnStream(const ExecuteParams& params) override;
|
| D | outfeed_thunk.h | 40 Status ExecuteOnStream(const ExecuteParams& params) override;
|
| D | replica_id_thunk.cc | 21 Status ReplicaOrPartitionIdThunk::ExecuteOnStream(const ExecuteParams& params) { in ExecuteOnStream() function in xla::gpu::ReplicaOrPartitionIdThunk
|
| D | collective_permute_thunk.h | 35 Status ExecuteOnStream(const ExecuteParams& params) override;
|
| D | custom_call_thunk.h | 46 Status ExecuteOnStream(const ExecuteParams& params) override;
|
| D | convolution_thunk.h | 53 Status ExecuteOnStream(const ExecuteParams& params) override;
|
| D | for_thunk.h | 42 Status ExecuteOnStream(const ExecuteParams& params) override;
|
| D | sequential_thunk.h | 44 Status ExecuteOnStream(const ExecuteParams& params) override;
|
| D | replica_id_thunk.h | 28 Status ExecuteOnStream(const ExecuteParams& params) override;
|
| D | while_thunk.h | 53 Status ExecuteOnStream(const ExecuteParams& params) override;
|
| D | kernel_thunk.h | 62 Status ExecuteOnStream(const ExecuteParams& params) override;
|
| D | cholesky_thunk.h | 53 Status ExecuteOnStream(const ExecuteParams& params) override;
|
| D | triangular_solve_thunk.h | 51 Status ExecuteOnStream(const ExecuteParams& params) override;
|
| D | conditional_thunk.h | 67 Status ExecuteOnStream(const ExecuteParams& params) override;
|
| /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
140 ExecuteOnStream(&run_options[0], arguments[0], in ExecuteOnStreams()
|