| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| batch_norm_op.cc | 30 OP_REQUIRES_OK(ctx, ctx->GetAttr("is_training", &is_training_));
64 if (is_training_) {
97 bool is_training_; member in class:tensorflow::__anon38725::FusedBatchNormOp
107 OP_REQUIRES_OK(ctx, ctx->GetAttr("is_training", &is_training_));
149 if (is_training_) {
198 bool is_training_; member in class:tensorflow::__anon38725::FusedBatchNormGradOp
|
| /external/tensorflow/tensorflow/core/kernels/ |
| mkl_fused_batch_norm_op.cc | 58 OP_REQUIRES_OK(context, context->GetAttr("is_training", &is_training_));
92 if (is_training_) {
103 is_training_ ? dnnUseScaleShift
163 if (is_training_)
195 if (is_training_) {
212 bool is_training_; member in class:tensorflow::MklFusedBatchNormOp
700 OP_REQUIRES_OK(context, context->GetAttr("is_training", &is_training_));
934 bool is_training_; member in class:tensorflow::MklFusedBatchNormOp
1284 bool is_training_; member in class:tensorflow::MklFusedBatchNormGradOp
[all...] |
| fused_batch_norm_op.cc | 513 OP_REQUIRES_OK(context, context->GetAttr("is_training", &is_training_));
539 if (is_training_) {
569 tensor_format_, is_training_);
575 bool is_training_; member in class:tensorflow::FusedBatchNormOp
590 OP_REQUIRES_OK(context, context->GetAttr("is_training", &is_training_));
656 if (is_training_) {
686 bool is_training_; member in class:tensorflow::FusedBatchNormGradOp
[all...] |
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/kernels/ |
| cudnn_rnn_ops.cc | 803 OP_REQUIRES_OK(context, context->GetAttr("is_training", &is_training_));
918 bool is_training_; member in class:tensorflow::TFRNNInputMode::CudnnRNNParamsToCanonical::CudnnRNNForwardOp
[all...] |
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| layout_optimizer_test.cc | 154 attrs.is_training_ = is_training;
[all...] |
