Searched defs:variance_data (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | fold_old_batch_norms_test.cc | 65 Tensor variance_data(DT_FLOAT, TensorShape({2})); in TestFoldOldBatchNorms() local
149 Tensor variance_data(DT_FLOAT, TensorShape({4})); in TestFoldOldBatchNormsAfterDepthwiseConv2dNative() local
227 Tensor variance_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNorms() local
304 Tensor variance_data(DT_FLOAT, TensorShape({4})); in TestFoldFusedBatchNormsAfterDepthwiseConv2dNative() local
405 Tensor variance_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNormsWithConcat() local
497 Tensor variance_data(DT_FLOAT, TensorShape({2})); in TestFoldFusedBatchNormsWithBatchToSpace() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | conv_ops_test.cc | 612 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/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_fused_batch_norm_op.cc | 86 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()
942 auto variance_data = reinterpret_cast<U*>(saved_variance_data_tf); in Compute() local
1354 U* variance_data = static_cast<U*>( in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | fused_batchnorm_reserve_space_test.cc | 107 Tensor variance_data(DT_FLOAT, TensorShape({0})); in TEST() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | dnn.h | 2468 const DeviceMemory<float>& variance_data, double epsilon, in DoFusedBatchNormActivationInference()
2481 const DeviceMemory<float>& variance_data, double epsilon, in DoFusedBatchNormActivationInference()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/ |
| H A D | batch_normalization_test.cc | 702 std::unique_ptr<GlobalData> variance_data = in XLA_TEST_P() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/ |
| H A D | rocm_dnn.cc | 4670 const DeviceMemory<U>& mean_data, const DeviceMemory<U>& variance_data, in DoFusedBatchNormActivationInferenceImpl()
4747 const DeviceMemory<float>& variance_data, double epsilon, in DoFusedBatchNormActivationInference()
4763 const DeviceMemory<float>& variance_data, double epsilon, in DoFusedBatchNormActivationInference()
|