| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | replica_id_thunk.h | 31 ReplicaOrPartitionIdThunk(Kind kind, ThunkInfo thunk_info, in ReplicaOrPartitionIdThunk()
41 ReplicaIdThunk(ThunkInfo thunk_info, const BufferAllocation::Slice& dest) in ReplicaIdThunk()
47 PartitionIdThunk(ThunkInfo thunk_info, const BufferAllocation::Slice& dest) in PartitionIdThunk()
|
| D | memset_thunk.h | 35 explicit MemzeroThunk(ThunkInfo thunk_info, in MemzeroThunk()
49 explicit Memset32BitValueThunk(ThunkInfo thunk_info, uint32 value, in Memset32BitValueThunk()
|
| D | copy_thunk.cc | 25 ThunkInfo thunk_info, const void* source_address, in HostToDeviceCopyThunk()
42 ThunkInfo thunk_info, const BufferAllocation::Slice& source_buffer, in DeviceToDeviceCopyThunk()
|
| D | cudnn_batchnorm_thunk.cc | 35 ThunkInfo thunk_info, CudnnBatchNormConfig config, in CudnnBatchNormForwardInferenceThunk()
74 ThunkInfo thunk_info, CudnnBatchNormConfig config, in CudnnBatchNormForwardTrainingThunk()
118 ThunkInfo thunk_info, CudnnBatchNormConfig config, in CudnnBatchNormBackwardThunk()
|
| D | for_thunk.cc | 26 ForThunk::ForThunk(ThunkInfo thunk_info, const int64 loop_limit, in ForThunk()
|
| D | custom_call_thunk.cc | 30 CustomCallThunk::CustomCallThunk(ThunkInfo thunk_info, void* call_target, in CustomCallThunk()
|
| D | sequential_thunk.cc | 27 SequentialThunk::SequentialThunk(ThunkInfo thunk_info, in SequentialThunk()
|
| D | convolution_thunk.cc | 34 ThunkInfo thunk_info, GpuConvConfig config, in ConvolutionThunk()
|
| D | infeed_thunk.cc | 29 InfeedThunk::InfeedThunk(ThunkInfo thunk_info, in InfeedThunk()
|
| D | while_thunk.cc | 27 ThunkInfo thunk_info, in WhileThunk()
|
| D | cholesky_thunk.cc | 34 CholeskyThunk::CholeskyThunk(ThunkInfo thunk_info, in CholeskyThunk()
|
| D | outfeed_thunk.cc | 28 OutfeedThunk::OutfeedThunk(ThunkInfo thunk_info, in OutfeedThunk()
|
| D | conditional_thunk.cc | 48 ThunkInfo thunk_info, ConditionalThunkConfig config, in ConditionalThunk()
|
| D | triangular_solve_thunk.cc | 35 ThunkInfo thunk_info, const TriangularSolveOptions& options, in TriangularSolveThunk()
|
| D | nccl_all_gather_thunk.cc | 62 ThunkInfo thunk_info, mlir::lmhlo::AllGatherOp op, int64 replica_count, in NcclAllGatherThunk()
|
| D | kernel_thunk.cc | 36 KernelThunk::KernelThunk(ThunkInfo thunk_info, in KernelThunk()
|
| D | nccl_all_to_all_thunk.cc | 54 ThunkInfo thunk_info, mlir::lmhlo::AllToAllOp op, int64 replica_count, in NcclAllToAllThunk()
|
| D | nccl_all_reduce_thunk.cc | 111 ThunkInfo thunk_info, mlir::lmhlo::AllReduceOp op, int64 replica_count, in NcclAllReduceThunk()
|
| D | fft_thunk.cc | 101 FftThunk::FftThunk(ThunkInfo thunk_info, FftType fft_type, in FftThunk()
|
| D | thunk.h | 82 explicit Thunk(Kind kind, ThunkInfo thunk_info) in Thunk() argument
|
| D | collective_permute_thunk.cc | 222 ThunkInfo thunk_info, in CollectivePermuteThunk()
|
| D | ir_emitter_unnested.h | 62 Thunk::ThunkInfo thunk_info; member
|
| D | gemm_thunk.cc | 46 GemmThunk::GemmThunk(ThunkInfo thunk_info, GpuGemmConfig config, in GemmThunk()
|
| D | ir_emitter_unnested.cc | 3342 absl::string_view name, Thunk::ThunkInfo thunk_info, in BuildKernelThunkFromBufferSlices()
3484 absl::string_view name, Thunk::ThunkInfo thunk_info, in BuildKernelThunkForMlirImpl()
3515 mlir::Operation* op, mlir::ValueRange operands, Thunk::ThunkInfo thunk_info, in BuildKernelThunkForMlir()
3537 mlir::Operation* op, Thunk::ThunkInfo thunk_info, in BuildKernelThunkForMlir()
|