| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | fold_old_batch_norms_test.cc | 65 Tensor variance_data(DT_FLOAT, TensorShape({2})); in TestFoldOldBatchNorms() local
66 test::FillValues<float>(&variance_data, {0.25f, 0.5f}); in TestFoldOldBatchNorms()
68 Input::Initializer(variance_data)); in TestFoldOldBatchNorms()
149 Tensor variance_data(DT_FLOAT, TensorShape({4})); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
150 test::FillValues<float>(&variance_data, {0.25f, 0.5f, 0.75f, 1.0f}); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
152 Input::Initializer(variance_data)); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative()
227 Tensor variance_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNorms() local
228 test::FillValues<float>(&variance_data, {0.25f, 0.5f}); in TestFoldFusedBatchNorms()
230 Input::Initializer(variance_data)); in TestFoldFusedBatchNorms()
304 Tensor variance_data(DT_FLOAT, TensorShape({4})); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
[all …]
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 93 Tensor variance_data(DT_FLOAT, TensorShape({0})); in TEST() local
95 Const(root.WithOpName("variance"), Input::Initializer(variance_data)); in TEST()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops_test.cc | 545 const Tensor& mean_data, const Tensor& variance_data, Tensor* out)>;
653 const Tensor& variance_data, Tensor* output, in RunConv2DWithBatchNorm() argument
671 ops::Const(root.WithOpName("var"), Input::Initializer(variance_data)), in RunConv2DWithBatchNorm()
680 const Tensor& mean_data, const Tensor& variance_data, Tensor* output, in RunConv2DWithBatchNormAndRelu() argument
698 ops::Const(root.WithOpName("var"), Input::Initializer(variance_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 | mkl_fused_batch_norm_op.cc | 64 T* mean_data, T* variance_data) { in Execute() argument
76 context_.variance_mem->set_data_handle(static_cast<void*>(variance_data)); in Execute()
306 void Execute(const T* src_data, const T* mean_data, const T* variance_data, in Execute() argument
314 static_cast<void*>(const_cast<T*>(variance_data))); in Execute()
697 auto variance_data = reinterpret_cast<T*>(saved_variance_data_tf); in Compute() local
701 batch_variance_data[k] = variance_data[k] * adjust_factor; in Compute()
704 std::memcpy(batch_variance_data, variance_data, depth_ * sizeof(T)); in Compute()
1006 T* variance_data = static_cast<T*>( in Compute() local
1012 bn_bwd->Execute(src_data, mean_data, variance_data, diff_dst_data, in Compute()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 552 const DeviceMemory<float>& variance_data, double epsilon,
563 const DeviceMemory<float>& variance_data, double epsilon,
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
4170 variance_data.opaque(), epsilon); in DoFusedBatchNormActivationInferenceImpl()
4212 const DeviceMemory<float>& variance_data, double epsilon, in DoFusedBatchNormActivationInference() argument
4218 variance_data, epsilon, activation_mode, y_data, output_profile_result); in DoFusedBatchNormActivationInference()
4228 const DeviceMemory<float>& variance_data, double epsilon, in DoFusedBatchNormActivationInference() argument
4234 variance_data, epsilon, activation_mode, y_data, output_profile_result); in DoFusedBatchNormActivationInference()
|
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | batch_normalization_test.cc | 639 std::unique_ptr<GlobalData> variance_data = in XLA_TEST_P() local
654 variance_data.get()}, in XLA_TEST_P()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.h | 2399 const DeviceMemory<float>& variance_data, double epsilon,
2412 const DeviceMemory<float>& variance_data, double epsilon,
|