Searched refs:num_side_inputs (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 49 int num_side_inputs; in FusedBatchNormOp() local
50 OP_REQUIRES_OK(ctx, ctx->GetAttr("num_side_inputs", &num_side_inputs)); in FusedBatchNormOp()
51 OP_REQUIRES(ctx, num_side_inputs >= 0 && num_side_inputs <= 1, in FusedBatchNormOp()
54 add_side_input_ = (num_side_inputs == 1); in FusedBatchNormOp()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 257 int num_side_inputs = 0; in RunFusedBatchNormEx() local
261 num_side_inputs = 1; in RunFusedBatchNormEx()
278 .Attr("num_side_inputs", num_side_inputs) in RunFusedBatchNormEx()
567 int num_side_inputs = 0; in FusedBatchNormEx() local
575 num_side_inputs = 1; in FusedBatchNormEx()
592 .Attr("num_side_inputs", num_side_inputs) in FusedBatchNormEx()
|
| D | fused_batch_norm_op.cc | 1206 int num_side_inputs; in FusedBatchNormOpBase() local
1208 context->GetAttr("num_side_inputs", &num_side_inputs)); in FusedBatchNormOpBase()
1209 OP_REQUIRES(context, num_side_inputs >= 0 && num_side_inputs <= 1, in FusedBatchNormOpBase()
1212 has_side_input_ = (num_side_inputs == 1); in FusedBatchNormOpBase()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 662 int num_side_inputs; in MklFusedBatchNormOp() local
664 context->GetAttr("num_side_inputs", &num_side_inputs)); in MklFusedBatchNormOp()
666 OP_REQUIRES(context, num_side_inputs == 0, in MklFusedBatchNormOp()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | mkl_layout_pass.cc | 1753 int num_side_inputs; in FusedBatchNormExRewrite() local
1754 TF_CHECK_OK(GetNodeAttr(n->def(), "num_side_inputs", &num_side_inputs)); in FusedBatchNormExRewrite()
1759 if (num_side_inputs != 0) { in FusedBatchNormExRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 18324 TF_DerivedOperandSizeAttr num_side_inputs = TF_DerivedOperandSizeAttr<5>;
|