Home
last modified time | relevance | path

Searched defs:kFeatureIndex (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc217 const int kFeatureIndex = 3; in XLA_TEST_F() local
240 const int kFeatureIndex = 3; in XLA_TEST_F() local
270 const int kFeatureIndex = 2; in XLA_TEST_F() local
294 const int kFeatureIndex = 2; in XLA_TEST_F() local
325 const int kFeatureIndex = 0; in XLA_TEST_F() local
356 const int kFeatureIndex = 2; in XLA_TEST_F() local
388 const int kFeatureIndex = 2; in XLA_TEST_F() local
417 const int kFeatureIndex = 2; in XLA_TEST_F() local
H A Dbfloat16_test.cc80 const int kFeatureIndex = 2; in XLA_TEST_F() local
116 const int kFeatureIndex = 2; in XLA_TEST_F() local