Searched refs:mean_data (Results 1 - 5 of 5) sorted by relevance
/external/tensorflow/tensorflow/tools/graph_transforms/ |
H A D | fold_old_batch_norms_test.cc | 59 Tensor mean_data(DT_FLOAT, TensorShape({2})); 60 test::FillValues<float>(&mean_data, {10.0f, 20.0f}); 62 Const(root.WithOpName("mean_op"), Input::Initializer(mean_data)); 143 Tensor mean_data(DT_FLOAT, TensorShape({2})); 144 test::FillValues<float>(&mean_data, {10.0f, 20.0f}); 146 Const(root.WithOpName("mean_op"), Input::Initializer(mean_data)); 243 Tensor mean_data(DT_FLOAT, TensorShape({2})); 244 test::FillValues<float>(&mean_data, {10.0f, 20.0f}); 246 Const(root.WithOpName("mean_op"), Input::Initializer(mean_data));
|
/external/tensorflow/tensorflow/core/kernels/ |
H A D | adjust_contrast_op.cc | 217 auto mean_data = mean_values.tensor<float, 2>(); variable 221 ReduceMeanAcrossImage(input_data, mean_data, output_data); 223 BroadcastAcrossImage(mean_data, output_data);
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
H A D | batch_normalization_test.cc | 627 std::unique_ptr<GlobalData> mean_data = local 643 {input_data.get(), scale_data.get(), offset_data.get(), mean_data.get(), 810 std::unique_ptr<GlobalData> mean_data = local 832 {input_data.get(), scale_data.get(), mean_data.get(),
|
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/reference/ |
H A D | reference_ops.h | 564 const float* input_data, const Dims<4>& input_dims, const float* mean_data, 586 mean_data[Offset(mean_dims, c, x, y, 0)]) * 597 const Dims<4>& input_dims, const float* mean_data, 617 mean_data[Offset(mean_dims, c, 0, 0, 0)]) * 563 NonGlobalBatchNormalization( const float* input_data, const Dims<4>& input_dims, const float* mean_data, const Dims<4>& mean_dims, const float* multiplier_data, const Dims<4>& multiplier_dims, const float* offset_data, const Dims<4>& offset_dims, float* output_data, const Dims<4>& output_dims) argument 596 GlobalBatchNormalization(const float* input_data, const Dims<4>& input_dims, const float* mean_data, const Dims<4>& mean_dims, const float* multiplier_data, const Dims<4>& multiplier_dims, const float* offset_data, const Dims<4>& offset_dims, float* output_data, const Dims<4>& output_dims) argument
|
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
H A D | optimized_ops.h | 1103 const float* input_data, const Dims<4>& input_dims, const float* mean_data, 1126 mean_data[Offset(mean_dims, c, x, y, 0)]) * 1137 const Dims<4>& input_dims, const float* mean_data, 1158 mean_data[Offset(mean_dims, c, 0, 0, 0)]) * 1102 NonGlobalBatchNormalization( const float* input_data, const Dims<4>& input_dims, const float* mean_data, const Dims<4>& mean_dims, const float* multiplier_data, const Dims<4>& multiplier_dims, const float* offset_data, const Dims<4>& offset_dims, float* output_data, const Dims<4>& output_dims) argument 1136 GlobalBatchNormalization(const float* input_data, const Dims<4>& input_dims, const float* mean_data, const Dims<4>& mean_dims, const float* multiplier_data, const Dims<4>& multiplier_dims, const float* offset_data, const Dims<4>& offset_dims, float* output_data, const Dims<4>& output_dims) argument
|
Completed in 538 milliseconds