Home
last modified time | relevance | path

Searched refs:num_side_inputs (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc53 int num_side_inputs; in FusedBatchNormOp() local
54 OP_REQUIRES_OK(ctx, ctx->GetAttr("num_side_inputs", &num_side_inputs)); in FusedBatchNormOp()
55 OP_REQUIRES(ctx, num_side_inputs >= 0 && num_side_inputs <= 1, in FusedBatchNormOp()
58 add_side_input_ = (num_side_inputs == 1); in FusedBatchNormOp()
/external/tensorflow/tensorflow/core/kernels/
Dfused_batch_norm_ex_op_test.cc256 int num_side_inputs = 0; in RunFusedBatchNormEx() local
260 num_side_inputs = 1; in RunFusedBatchNormEx()
277 .Attr("num_side_inputs", num_side_inputs) in RunFusedBatchNormEx()
301 .Attr("num_side_inputs", num_side_inputs) in RunFusedBatchNormEx()
588 int num_side_inputs = 0; in FusedBatchNormEx() local
596 num_side_inputs = 1; in FusedBatchNormEx()
613 .Attr("num_side_inputs", num_side_inputs) in FusedBatchNormEx()
Dfused_batch_norm_op.cc1232 int num_side_inputs; in FusedBatchNormOpBase() local
1234 context->GetAttr("num_side_inputs", &num_side_inputs)); in FusedBatchNormOpBase()
1235 OP_REQUIRES(context, num_side_inputs >= 0 && num_side_inputs <= 1, in FusedBatchNormOpBase()
1238 has_side_input_ = (num_side_inputs == 1); in FusedBatchNormOpBase()
1458 int num_side_inputs; in FusedBatchNormGradOpBase() local
1460 context->GetAttr("num_side_inputs", &num_side_inputs)); in FusedBatchNormGradOpBase()
1461 OP_REQUIRES(context, num_side_inputs >= 0 && num_side_inputs <= 1, in FusedBatchNormGradOpBase()
1464 has_side_input_ = (num_side_inputs == 1); in FusedBatchNormGradOpBase()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_fused_batch_norm_op.cc686 int num_side_inputs; in MklFusedBatchNormOp() local
688 context->GetAttr("num_side_inputs", &num_side_inputs)); in MklFusedBatchNormOp()
690 OP_REQUIRES(context, num_side_inputs == 0, in MklFusedBatchNormOp()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc1303 int num_side_inputs; in FusedBatchNormGradExShape() local
1304 TF_RETURN_IF_ERROR(c->GetAttr("num_side_inputs", &num_side_inputs)); in FusedBatchNormGradExShape()
1305 if (num_side_inputs == 0) { in FusedBatchNormGradExShape()
/external/tensorflow/tensorflow/core/common_runtime/
Dmkl_layout_pass.cc1705 int num_side_inputs; in FusedBatchNormExRewrite() local
1706 TF_CHECK_OK(GetNodeAttr(n->def(), "num_side_inputs", &num_side_inputs)); in FusedBatchNormExRewrite()
1711 if (num_side_inputs != 0) { in FusedBatchNormExRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_generated_ops.td20862 TF_DerivedOperandSizeAttr num_side_inputs = TF_DerivedOperandSizeAttr<5>;