HomeSort by relevance Sort by last modified time
    Searched refs:mean_square (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/tests/
batch_normalization_test.cc 478 std::vector<float> mean_square(feature_bound);
480 mean_square[i] = mean[i] * mean[i];
490 var[i] = square_mean[i] - mean_square[i];
578 std::vector<float> mean_square(feature_bound);
580 mean_square[i] = mean[i] * mean[i];
590 var[i] = square_mean[i] - mean_square[i];
693 std::vector<float> mean_square(feature_bound);
695 mean_square[i] = mean[i] * mean[i];
709 var[i] = square_mean[i] - mean_square[i];
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
batchnorm_expander.cc 243 auto mean_square = add(HloInstruction::CreateBinary( local
248 feature_shape, HloOpcode::kSubtract, square_mean, mean_square));
  /external/tensorflow/tensorflow/go/op/
wrappers.go     [all...]

Completed in 320 milliseconds