Searched refs:AssignCommonParams (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_batchnorm_runner.cc | 113 void AssignCommonParams(const CudnnBatchNormConfig& config, in AssignCommonParams() function
232 AssignCommonParams(config, &inference_params.common, operand, scale); in RunCudnnBatchNormForwardInference()
261 AssignCommonParams(config, &forward_params.common, operand, scale); in RunCudnnBatchNormForwardTraining()
292 AssignCommonParams(config, &backward_params.common, operand, scale); in RunCudnnBatchNormBackward()
|