Searched refs:feature_bound (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc451 const int64 feature_bound = bounds[feature_index]; local
452 std::vector<float> offset(feature_bound, 1);
453 std::vector<float> scale(feature_bound, 2);
472 std::vector<float> mean(feature_bound);
474 for (int64 i = 0; i < feature_bound; ++i) {
478 std::vector<float> mean_square(feature_bound);
479 for (int64 i = 0; i < feature_bound; ++i) {
483 std::vector<float> square_mean(feature_bound);
484 for (int64 i = 0; i < feature_bound; ++i) {
488 std::vector<float> var(feature_bound);
551 const int64 feature_bound = bounds[feature_index]; local
663 const int64 feature_bound = bounds[feature_index]; local
[all...]

Completed in 142 milliseconds