Searched refs:output_inv_stddev (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_thunk.cc | 79 const BufferAllocation::Slice& output_inv_stddev) in CudnnBatchNormForwardTrainingThunk() argument
87 output_inv_stddev_(output_inv_stddev) {} in CudnnBatchNormForwardTrainingThunk()
98 se::DeviceMemory<float> output_inv_stddev( in ExecuteOnStream() local
106 config_, operand, output_data, output_mean, output_inv_stddev, in ExecuteOnStream()
|
| D | cudnn_batchnorm_runner.cc | 52 se::DeviceMemory<float> output_inv_stddev; member
183 /*saved_inv_var=*/¶ms->output_inv_stddev, // in RunCudnnBatchNormForwardTrainingImpl()
258 se::DeviceMemory<float> output_inv_stddev, se::DeviceMemory<float> scale, in RunCudnnBatchNormForwardTraining() argument
265 forward_params.output_inv_stddev = output_inv_stddev; in RunCudnnBatchNormForwardTraining()
|
| D | cudnn_batchnorm_runner.h | 51 se::DeviceMemory<float> output_inv_stddev, se::DeviceMemory<float> scale,
|
| D | cudnn_batchnorm_thunk.h | 85 const BufferAllocation::Slice& output_inv_stddev);
|
| D | ir_emitter_unnested.cc | 1375 TF_ASSIGN_OR_RETURN(auto output_inv_stddev, in EmitBatchNormThunkFromMlir()
1387 /*output_inv_stddev=*/output_inv_stddev)); in EmitBatchNormThunkFromMlir()
|