Searched refs:bn_inference (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1443 if (auto bn_inference = in EmitBatchNormThunkFromMlir() local
1447 {{bn_inference.scale(), "scale"}, in EmitBatchNormThunkFromMlir()
1448 {bn_inference.offset(), "offset"}, in EmitBatchNormThunkFromMlir()
1449 {bn_inference.mean(), "mean"}, in EmitBatchNormThunkFromMlir()
1450 {bn_inference.stddev(), "stddev"}}, in EmitBatchNormThunkFromMlir()
1452 {{bn_inference.operand(), "operand"}, in EmitBatchNormThunkFromMlir()
1453 {bn_inference.output(), "output"}})); in EmitBatchNormThunkFromMlir()
1456 GetAllocationSliceForMlir(bn_inference.operand())); in EmitBatchNormThunkFromMlir()
1458 GetAllocationSliceForMlir(bn_inference.scale())); in EmitBatchNormThunkFromMlir()
1460 GetAllocationSliceForMlir(bn_inference.offset())); in EmitBatchNormThunkFromMlir()
[all …]
|