| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_old_batch_norms_test.cc | 60 Tensor mean_data(DT_FLOAT, TensorShape({2})); in TestFoldOldBatchNorms() local
61 test::FillValues<float>(&mean_data, {10.0f, 20.0f}); in TestFoldOldBatchNorms()
63 Const(root.WithOpName("mean_op"), Input::Initializer(mean_data)); in TestFoldOldBatchNorms()
144 Tensor mean_data(DT_FLOAT, TensorShape({4})); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
145 test::FillValues<float>(&mean_data, {10.0f, 20.0f, 30.0f, 40.0f}); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
147 Const(root.WithOpName("mean_op"), Input::Initializer(mean_data)); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
222 Tensor mean_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNorms() local
223 test::FillValues<float>(&mean_data, {10.0f, 20.0f}); in TestFoldFusedBatchNorms()
225 Const(root.WithOpName("mean_op"), Input::Initializer(mean_data)); in TestFoldFusedBatchNorms()
299 Tensor mean_data(DT_FLOAT, TensorShape({4})); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 90 Tensor mean_data(DT_FLOAT, TensorShape({0})); in TEST() local
91 Output mean = Const(root.WithOpName("offset"), Input::Initializer(mean_data)); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_test.cc | 545 const Tensor& mean_data, const Tensor& variance_data, Tensor* out)>;
652 const Tensor& mean_data, in RunConv2DWithBatchNorm() argument
670 ops::Const(root.WithOpName("mean"), Input::Initializer(mean_data)), in RunConv2DWithBatchNorm()
680 const Tensor& mean_data, const Tensor& variance_data, Tensor* output, in RunConv2DWithBatchNormAndRelu() argument
697 ops::Const(root.WithOpName("mean"), Input::Initializer(mean_data)), in RunConv2DWithBatchNormAndRelu()
894 const Tensor& mean_data, const Tensor& variance_data, in VerifyConv2DWithBatchNorm()
897 offset_data, mean_data, variance_data, out); in VerifyConv2DWithBatchNorm()
903 const Tensor& mean_data, const Tensor& variance_data, in VerifyConv2DWithBatchNorm() argument
906 {scale_data, offset_data, mean_data, variance_data}, in VerifyConv2DWithBatchNorm()
924 const Tensor& mean_data, const Tensor& variance_data, in VerifyConv2DWithBatchNormAndRelu() argument
[all …]
|
| D | adjust_contrast_op.cc | 217 auto mean_data = mean_values.tensor<float, 2>(); in DoCompute() local
221 ReduceMeanAcrossImage(input_data, mean_data, output_data); in DoCompute()
223 BroadcastAcrossImage(mean_data, output_data); in DoCompute()
|
| D | mkl_fused_batch_norm_op.cc | 64 T* mean_data, T* variance_data) { in Execute() argument
75 context_.mean_mem->set_data_handle(static_cast<void*>(mean_data)); in Execute()
306 void Execute(const T* src_data, const T* mean_data, const T* variance_data, in Execute() argument
312 static_cast<void*>(const_cast<T*>(mean_data))); in Execute()
1004 T* mean_data = in Compute() local
1012 bn_bwd->Execute(src_data, mean_data, variance_data, diff_dst_data, in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 637 std::unique_ptr<GlobalData> mean_data = in XLA_TEST_P() local
653 {input_data.get(), scale_data.get(), offset_data.get(), mean_data.get(), in XLA_TEST_P()
820 std::unique_ptr<GlobalData> mean_data = in XLA_TEST_P() local
840 {input_data.get(), scale_data.get(), mean_data.get(), in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 551 const DeviceMemory<float>& mean_data,
562 const DeviceMemory<float>& mean_data,
725 const DeviceMemory<U>& mean_data, const DeviceMemory<U>& variance_data,
|
| D | rocm_dnn.cc | 4135 const DeviceMemory<U>& mean_data, const DeviceMemory<U>& variance_data, in DoFusedBatchNormActivationInferenceImpl() argument
4169 scale_data.opaque(), offset_data.opaque(), mean_data.opaque(), in DoFusedBatchNormActivationInferenceImpl()
4211 const DeviceMemory<float>& mean_data, in DoFusedBatchNormActivationInference() argument
4217 scale_offset_mean_variance_descriptor, scale_data, offset_data, mean_data, in DoFusedBatchNormActivationInference()
4227 const DeviceMemory<float>& mean_data, in DoFusedBatchNormActivationInference() argument
4233 scale_offset_mean_variance_descriptor, scale_data, offset_data, mean_data, in DoFusedBatchNormActivationInference()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 2398 const DeviceMemory<float>& mean_data,
2411 const DeviceMemory<float>& mean_data,
|