Home
last modified time | relevance | path

Searched defs:batch_variance (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfused_batch_norm_op.cc186 Eigen::Tensor<U, 1, Eigen::RowMajor> batch_variance(depth); in operator ()() local
303 typename TTypes<U>::Vec batch_variance(batch_var_output->vec<U>()); in operator ()() local
H A Dfused_batch_norm_ex_op_test.cc53 Tensor batch_variance; member
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A Dlegalize_tf.cc2389 Value batch_variance = bn_train_op.getResult(2); in matchAndRewrite() local