Home
last modified time | relevance | path

Searched defs:mean_data (Results 1 – 20 of 20) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dfold_old_batch_norms_test.cc60 Tensor mean_data(DT_FLOAT, TensorShape({2})); in TestFoldOldBatchNorms() local
144 Tensor mean_data(DT_FLOAT, TensorShape({4})); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
222 Tensor mean_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNorms() local
299 Tensor mean_data(DT_FLOAT, TensorShape({4})); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
400 Tensor mean_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNormsWithConcat() local
492 Tensor mean_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNormsWithBatchToSpace() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A DNormalization.cpp30 const float* mean_data, in compute_fused_params()
94 const float* mean_data = mean.template const_data_ptr<float>(); in q_batch_norm1d_impl() local
203 const float* mean_data = mean.template const_data_ptr<float>(); in q_batch_norm2d_impl() local
299 const float* mean_data = mean.template const_data_ptr<float>(); in q_batch_norm3d_impl() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A Dlayer_norm_kernel.cpp43 T* mean_data = mean ? mean->data_ptr<T>() : nullptr; in LayerNormKernelImplInternal() local
103 param_t* mean_data = mean ? mean->data_ptr<param_t>() : nullptr; in layer_norm_kernel_mixed_type() local
188 const T2* mean_data, in layer_norm_backward_frame()
303 const float* mean_data, in layer_norm_backward_frame()
510 const T2* mean_data = mean.template const_data_ptr<T2>(); in LayerNormBackwardKernelImplInternal() local
H A Dbatch_norm_kernel.cpp189 scalar_t* mean_data = mean.data_ptr<scalar_t>(); in batch_norm_cpu_collect_stats_contiguous_impl() local
233 scalar_t* mean_data = mean.data_ptr<scalar_t>(); in batch_norm_cpu_collect_stats_channels_last_impl() local
841 param_t* mean_data = mean.data_ptr<param_t>(); in batch_norm_cpu_collect_stats_contiguous_internal() local
912 param_t* mean_data = mean.data_ptr<param_t>(); in batch_norm_cpu_collect_stats_channels_last_internal() local
1150 opmath_t* mean_data = mean.data_ptr<opmath_t>(); in batch_norm_cpu_backward_channels_last_internal() local
H A Dgroup_norm_kernel.cpp50 PT* mean_data = mean.data_ptr<PT>(); in GroupNormKernelImplInternal() local
305 PT* mean_data = mean.data_ptr<PT>(); in GroupNormKernelImplChannelsLastInternal() local
902 const PT* mean_data = mean.const_data_ptr<PT>(); in GroupNormBackwardKernelImplInternal() local
1382 const PT* mean_data = mean.const_data_ptr<PT>(); in GroupNormBackwardKernelImplChannelsLastInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfused_batch_norm_ex_op_test.cc138 const Tensor& offset_data, const Tensor& mean_data, in RunFusedBatchNorm()
225 const Tensor& offset_data, const Tensor& mean_data, in RunFusedBatchNormEx()
467 const Tensor& mean_data, const Tensor& var_data, in VerifyFusedBatchNormEx()
469 FusedBatchNormGradOutputs* bwd) { in VerifyFusedBatchNormEx()
H A Dconv_ops_test.cc612 const Tensor& mean_data, const Tensor& variance_data, in RunConv2DWithBatchNorm()
641 const Tensor& mean_data, const Tensor& variance_data, in RunConv2DWithBatchNormAndActivation()
888 const Tensor& mean_data, const Tensor& variance_data, Tensor* out) { in __anona74cd8200602()
898 const Tensor& mean_data, const Tensor& variance_data, Tensor* out) { in __anona74cd8200702()
921 const Tensor& mean_data, const Tensor& variance_data, Tensor* out) { in __anona74cd8200802()
931 const Tensor& mean_data, const Tensor& variance_data, Tensor* out) { in __anona74cd8200902()
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_native_batch_norm.cpp111 const CTYPE* const mean_data = running_mean.const_data_ptr<CTYPE>(); in _native_batch_norm_legit_no_training_out() local
267 CTYPE* mean_data = mean_out.mutable_data_ptr<CTYPE>(); in _native_batch_norm_legit_no_stats_out() local
H A Dop_native_layer_norm.cpp44 CTYPE* mean_data = mean.mutable_data_ptr<CTYPE>(); in layer_norm() local
H A Dop_native_group_norm.cpp50 CTYPE* mean_data = mean.mutable_data_ptr<CTYPE>(); in group_norm() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_fused_batch_norm_op.cc86 U* mean_data, U* variance_data, in Execute()
437 void Execute(const T* src_data, const U* mean_data, const U* variance_data, in Execute()
941 auto mean_data = reinterpret_cast<U*>(saved_mean_data_tf); in Compute() local
1352 U* mean_data = in Compute() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A Dgroup_norm_kernel.cu576 T* mean_data = mean.mutable_data_ptr<T>(); in GroupNormKernelImplInternal() local
687 const T* mean_data = mean.const_data_ptr<T>(); in GroupNorm1dBackward() local
831 const T* mean_data = mean.const_data_ptr<T>(); in GroupNormBackwardKernelImplInternal() local
H A Dlayer_norm_kernel.cu741 T_ACC* mean_data, in launch_vectorized_layer_norm_kernel()
775 T_ACC* mean_data = mean->data_ptr<T_ACC>(); in LayerNormKernelImplInternal() local
1154 const T_ACC* mean_data = mean.template const_data_ptr<T_ACC>(); in LayerNormBackwardKernelImplInternal() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dfused_batchnorm_reserve_space_test.cc104 Tensor mean_data(DT_FLOAT, TensorShape({0})); in TEST() local
/aosp_15_r20/external/executorch/kernels/optimized/cpu/
H A Dop_native_layer_norm.cpp49 CTYPE* mean_data = mean.mutable_data_ptr<CTYPE>(); in layer_norm() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dbatch_normalization_test.cc700 std::unique_ptr<GlobalData> mean_data = in XLA_TEST_P() local
883 std::unique_ptr<GlobalData> mean_data = in XLA_TEST_P() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dadjust_contrast_op.cc226 auto mean_data = mean_values.tensor<float, 2>(); in DoCompute() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddnn.h2467 const DeviceMemory<float>& mean_data, in DoFusedBatchNormActivationInference()
2480 const DeviceMemory<float>& mean_data, in DoFusedBatchNormActivationInference()
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_dnn.cc4670 const DeviceMemory<U>& mean_data, const DeviceMemory<U>& variance_data, in DoFusedBatchNormActivationInferenceImpl()
4746 const DeviceMemory<float>& mean_data, in DoFusedBatchNormActivationInference()
4762 const DeviceMemory<float>& mean_data, in DoFusedBatchNormActivationInference()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/kernels/
H A DQuantizedOpKernels.cpp3195 float* mean_data = mean.mutable_data_ptr<float>(); local