Searched refs:CudnnBatchNormConfig (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_runner.h | 31 struct CudnnBatchNormConfig { struct
38 CudnnBatchNormConfig GetCudnnBatchNormConfig(const HloInstruction *instr, argument
43 const CudnnBatchNormConfig &config, se::DeviceMemoryBase operand,
49 const CudnnBatchNormConfig &config, se::DeviceMemoryBase operand,
55 const CudnnBatchNormConfig &config, se::DeviceMemoryBase operand,
|
| D | cudnn_batchnorm_thunk.h | 51 CudnnBatchNormConfig config,
67 CudnnBatchNormConfig config_;
79 ThunkInfo thunk_info, CudnnBatchNormConfig config,
95 CudnnBatchNormConfig config_;
107 ThunkInfo thunk_info, CudnnBatchNormConfig config,
123 const CudnnBatchNormConfig config_;
|
| D | cudnn_batchnorm_runner.cc | 113 void AssignCommonParams(const CudnnBatchNormConfig& config, in AssignCommonParams()
212 CudnnBatchNormConfig GetCudnnBatchNormConfig(const HloInstruction* instr, in GetCudnnBatchNormConfig()
215 CudnnBatchNormConfig config; in GetCudnnBatchNormConfig()
227 const CudnnBatchNormConfig& config, se::DeviceMemoryBase operand, in RunCudnnBatchNormForwardInference()
256 const CudnnBatchNormConfig& config, se::DeviceMemoryBase operand, in RunCudnnBatchNormForwardTraining()
285 const CudnnBatchNormConfig& config, se::DeviceMemoryBase operand, in RunCudnnBatchNormBackward()
|
| 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 | ir_emitter_unnested.cc | 1341 CudnnBatchNormConfig config; in EmitBatchNormThunkFromMlir()
1423 CudnnBatchNormConfig config; in EmitBatchNormThunkFromMlir()
|