Home
last modified time | relevance | path

Searched defs:num_side_inputs (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfused_batch_norm_ex_op_test.cc256 int num_side_inputs = 0; in RunFusedBatchNormEx() local
588 int num_side_inputs = 0; in FusedBatchNormEx() local
H A Dfused_batch_norm_op.cc1232 int num_side_inputs; in FusedBatchNormOpBase() local
1458 int num_side_inputs; in FusedBatchNormGradOpBase() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
H A Dbatch_norm_op.cc53 int num_side_inputs; in FusedBatchNormOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_fused_batch_norm_op.cc686 int num_side_inputs; in MklFusedBatchNormOp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc1303 int num_side_inputs; in FusedBatchNormGradExShape() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dmkl_layout_pass.cc1705 int num_side_inputs; in FusedBatchNormExRewrite() local