Searched refs:bn_grad (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1391 if (auto bn_grad = in EmitBatchNormThunkFromMlir() local
1395 {{bn_grad.scale(), "scale"}, in EmitBatchNormThunkFromMlir()
1396 {bn_grad.mean(), "mean"}, in EmitBatchNormThunkFromMlir()
1397 {bn_grad.stddev(), "stddev"}, in EmitBatchNormThunkFromMlir()
1398 {bn_grad.grad_scale(), "grad_scale"}, in EmitBatchNormThunkFromMlir()
1399 {bn_grad.grad_offset(), "grad_offset"}}, in EmitBatchNormThunkFromMlir()
1401 {{bn_grad.operand(), "operand"}, in EmitBatchNormThunkFromMlir()
1402 {bn_grad.grad_output(), "grad_output"}, in EmitBatchNormThunkFromMlir()
1403 {bn_grad.grad_operand(), "grad_operand"}})); in EmitBatchNormThunkFromMlir()
1406 GetAllocationSliceForMlir(bn_grad.operand())); in EmitBatchNormThunkFromMlir()
[all …]
|