Searched defs:scale4D (Results 1 - 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | batch_normalization_test.cc | 496 auto scale4D = *ReferenceUtil::Broadcast1DTo4D(scale, bounds, feature_index); local
501 scale4D, offset4D, epsilon);
596 auto scale4D = *ReferenceUtil::Broadcast1DTo4D(scale, bounds, feature_index); local
601 scale4D, offset4D, epsilon);
715 auto scale4D = *ReferenceUtil::Broadcast1DTo4D(scale, bounds, feature_index); local
747 scale4D, rsqrt_var_add_epsilon, [](float a, float b) { return a * b; });
779 grad_activation, scale4D, [](float a, float b) { return a * b; });
|
Completed in 144 milliseconds