Searched defs:var4D (Results 1 - 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | batch_normalization_test.cc | 495 auto var4D = *ReferenceUtil::Broadcast1DTo4D(var, bounds, feature_index); local
500 auto normalized = *ReferenceUtil::BatchNorm4D(input_array, mean4D, var4D,
595 auto var4D = *ReferenceUtil::Broadcast1DTo4D(var, bounds, feature_index); local
600 auto normalized = *ReferenceUtil::BatchNorm4D(input_array, mean4D, var4D,
714 auto var4D = *ReferenceUtil::Broadcast1DTo4D(var, bounds, feature_index); local
718 var4D, [epsilon](float a) { return a + epsilon; });
|
Completed in 337 milliseconds