Searched refs:feature_bound (Results 1 – 1 of 1) sorted by relevance
460 const int64 feature_bound = bounds[feature_index]; in XLA_TEST_P() local461 std::vector<float> offset(feature_bound, 1); in XLA_TEST_P()462 std::vector<float> scale(feature_bound, 2); in XLA_TEST_P()481 std::vector<float> mean(feature_bound); in XLA_TEST_P()483 for (int64 i = 0; i < feature_bound; ++i) { in XLA_TEST_P()487 std::vector<float> mean_square(feature_bound); in XLA_TEST_P()488 for (int64 i = 0; i < feature_bound; ++i) { in XLA_TEST_P()492 std::vector<float> square_mean(feature_bound); in XLA_TEST_P()493 for (int64 i = 0; i < feature_bound; ++i) { in XLA_TEST_P()497 std::vector<float> var(feature_bound); in XLA_TEST_P()[all …]