Searched refs:bn_train (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_unnested.cc | 1352 if (auto bn_train = in EmitBatchNormThunkFromMlir() local
1356 {{bn_train.scale(), "scale"}, in EmitBatchNormThunkFromMlir()
1357 {bn_train.offset(), "offset"}, in EmitBatchNormThunkFromMlir()
1358 {bn_train.batch_mean(), "batch_mean"}, in EmitBatchNormThunkFromMlir()
1359 {bn_train.batch_stddev(), "batch_stddev"}}, in EmitBatchNormThunkFromMlir()
1361 {{bn_train.operand(), "operand"}, {bn_train.output(), "output"}})); in EmitBatchNormThunkFromMlir()
1363 GetAllocationSliceForMlir(bn_train.operand())); in EmitBatchNormThunkFromMlir()
1365 GetAllocationSliceForMlir(bn_train.scale())); in EmitBatchNormThunkFromMlir()
1367 GetAllocationSliceForMlir(bn_train.offset())); in EmitBatchNormThunkFromMlir()
1372 GetAllocationSliceForMlir(bn_train.output())); in EmitBatchNormThunkFromMlir()
[all …]
|