| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | batch_norm_op.cc | 39 OP_REQUIRES_OK(ctx, ctx->GetAttr("is_training", &is_training_)); in FusedBatchNormOp()
93 if (is_training_) { in CompileImpl()
198 bool is_training_; member in tensorflow::__anonfc91c4b10111::FusedBatchNormOp
242 OP_REQUIRES_OK(ctx, ctx->GetAttr("is_training", &is_training_)); in FusedBatchNormGradOp()
274 if (is_training_) { in Compile()
346 bool is_training_; member in tensorflow::__anonfc91c4b10111::FusedBatchNormGradOp
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op.cc | 654 OP_REQUIRES_OK(context, context->GetAttr("is_training", &is_training_)); in MklFusedBatchNormOp()
776 MklBatchNormFwdParams fwdParams(src_dims, depth_, epsilon_, is_training_, in Compute()
809 if (is_training_) in Compute()
873 if (is_training_) { in Compute()
885 if (is_training_) { in Compute()
921 bool is_training_; member in tensorflow::MklFusedBatchNormOp
1057 OP_REQUIRES_OK(context, context->GetAttr("is_training", &is_training_)); in MklFusedBatchNormGradOp()
1221 is_training_, src_md, diff_dst_md); in Compute()
1308 bool is_training_; member in tensorflow::MklFusedBatchNormGradOp
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_op.cc | 1198 OP_REQUIRES_OK(context, context->GetAttr("is_training", &is_training_)); in FusedBatchNormOpBase()
1213 if (has_side_input_ && is_training_) { in FusedBatchNormOpBase()
1221 if (activation_mode_ != FbnActivationMode::kIdentity && is_training_) { in FusedBatchNormOpBase()
1294 context, !is_training_ || x.dim_size(3) % 4 == 0, in ComputeWithReservedSpace()
1317 if (is_training_) { in ComputeWithReservedSpace()
1340 bool is_training_; member in tensorflow::FusedBatchNormOpBase
1394 OP_REQUIRES_OK(context, context->GetAttr("is_training", &is_training_)); in FusedBatchNormGradOpBase()
1475 if (is_training_) { in ComputeWithReservedSpace()
1501 bool is_training_; member in tensorflow::FusedBatchNormGradOpBase
|
| D | cudnn_rnn_ops.cc | 161 is_training_(is_training), in CudnnRnnParameters()
190 std::to_string(is_training_), in ToString()
203 batch_size_, dir_count_, has_dropout_, is_training_, in get_data_as_tuple()
214 const bool is_training_; member in tensorflow::__anonebbbcdba0111::CudnnRnnParameters
1437 OP_REQUIRES_OK(context, context->GetAttr("is_training", &is_training_)); in CudnnRNNForwardOp()
1517 input_c, params, is_training_, output, output_h, output_c, in ComputeAndReturnAlgorithm()
1538 bool is_training() const { return is_training_; } in is_training()
1563 if (!is_training_) { in AllocateOutputs()
1571 bool is_training_; member in tensorflow::CudnnRNNForwardOp
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer_test.cc | 219 attrs.is_training_ = is_training; in SimpleFusedBatchNormGrad()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| D | convert_nodes_test.cc | 2137 attrs.is_training_ = is_training; in CreateFusedBatchNormOp()
|