Home
last modified time | relevance | path

Searched refs:is_training_ (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dbatch_norm_op.cc33 OP_REQUIRES_OK(ctx, ctx->GetAttr("is_training", &is_training_)); in FusedBatchNormOp()
61 if (is_training_) { in Compile()
105 bool is_training_; member in tensorflow::__anonfc91c4b10111::FusedBatchNormOp
116 OP_REQUIRES_OK(ctx, ctx->GetAttr("is_training", &is_training_)); in FusedBatchNormGradOp()
148 if (is_training_) { in Compile()
220 bool is_training_; member in tensorflow::__anonfc91c4b10111::FusedBatchNormGradOp
/external/tensorflow/tensorflow/core/kernels/
Dmkl_fused_batch_norm_op.cc507 OP_REQUIRES_OK(context, context->GetAttr("is_training", &is_training_)); in MklFusedBatchNormOp()
554 if (is_training_) { in Compute()
590 if (is_training_) in Compute()
640 MklBatchNormFwdParams fwdParams(src_dims, depth_, epsilon_, is_training_); in Compute()
691 if (is_training_) { in Compute()
699 if (is_training_) { in Compute()
719 bool is_training_; member in tensorflow::MklFusedBatchNormOp
830 OP_REQUIRES_OK(context, context->GetAttr("is_training", &is_training_)); in MklFusedBatchNormGradOp()
957 is_training_); in Compute()
1044 bool is_training_; member in tensorflow::MklFusedBatchNormGradOp
Dfused_batch_norm_op.cc534 OP_REQUIRES_OK(context, context->GetAttr("is_training", &is_training_)); in FusedBatchNormOp()
560 if (is_training_) { in Compute()
590 tensor_format_, is_training_); in Compute()
596 bool is_training_; member in tensorflow::FusedBatchNormOp
611 OP_REQUIRES_OK(context, context->GetAttr("is_training", &is_training_)); in FusedBatchNormGradOp()
677 if (is_training_) { in Compute()
707 bool is_training_; member in tensorflow::FusedBatchNormGradOp
Dcudnn_rnn_ops.cc161 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
1256 OP_REQUIRES_OK(context, context->GetAttr("is_training", &is_training_)); in CudnnRNNForwardOp()
1332 input_c, params, is_training_, output, output_h, output_c, in ComputeAndReturnAlgorithm()
1353 bool is_training() const { return is_training_; } in is_training()
1377 if (!is_training_) { in AllocateOutputs()
1385 bool is_training_; member in tensorflow::CudnnRNNForwardOp
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dlayout_optimizer_test.cc213 attrs.is_training_ = is_training; in SimpleFusedBatchNormGrad()