Searched refs:kFeatureIndex (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_layout_assignment_test.cc | 110 const int64 kFeatureIndex = 1; in TEST_F() local
120 ShapeUtil::MakeShape(F32, {shape.dimensions(kFeatureIndex)}); in TEST_F()
144 LiteralUtil::CreateR0<int64>(kFeatureIndex))); in TEST_F()
181 const int64 kFeatureIndex = 1; in TEST_F() local
189 ShapeUtil::MakeShape(F32, {shape.dimensions(kFeatureIndex)}); in TEST_F()
214 LiteralUtil::CreateR0<int64>(kFeatureIndex))); in TEST_F()
253 const int64 kFeatureIndex = 1; in TEST_F() local
261 ShapeUtil::MakeShape(F32, {shape.dimensions(kFeatureIndex)}); in TEST_F()
292 LiteralUtil::CreateR0<int64>(kFeatureIndex))); in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | bfloat16_test.cc | 82 const int kFeatureIndex = 2; in XLA_TEST_F() local
98 BatchNormTraining(operand, scale, offset, /*epsilon=*/0.001, kFeatureIndex); in XLA_TEST_F()
118 const int kFeatureIndex = 2; in XLA_TEST_F() local
141 /*epsilon=*/0.0, kFeatureIndex); in XLA_TEST_F()
|
| D | batch_normalization_test.cc | 232 const int kFeatureIndex = 3; in XLA_TEST_P() local
243 /*epsilon=*/0.001, kFeatureIndex); in XLA_TEST_P()
255 const int kFeatureIndex = 2; in XLA_TEST_P() local
267 /*epsilon=*/0.001, kFeatureIndex); in XLA_TEST_P()
280 const int kFeatureIndex = 0; in XLA_TEST_P() local
297 /*epsilon=*/1, kFeatureIndex); in XLA_TEST_P()
311 const int kFeatureIndex = 2; in XLA_TEST_P() local
329 /*epsilon=*/-100, kFeatureIndex); in XLA_TEST_P()
343 const int kFeatureIndex = 2; in XLA_TEST_P() local
360 /*epsilon=*/0.0, kFeatureIndex); in XLA_TEST_P()
|