Searched refs:dev_squares (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 154 auto dev_squares = Square(activation_deviations); in XLA_TEST_P() local
155 Reduce(dev_squares, ConstantR0<float>(&builder, 0.0f), add, {0, 2, 3}); in XLA_TEST_P()
199 auto dev_squares = Square(activation_deviations); in XLA_TEST_P() local
202 Reduce(dev_squares, ConstantR0<float>(&builder, 0.0f), add, in XLA_TEST_P()
|