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

/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dfold_old_batch_norms_test.cc59 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 Dadjust_contrast_op.cc217 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 Dbatch_normalization_test.cc627 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 Dreference_ops.h564 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 Doptimized_ops.h1103 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