| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | fused_batchnorm_reserve_space_test.cc | 116 scale, offset, mean, variance, FusedBatchNorm::Attrs{}.IsTraining(true)); in TEST()
119 scale, offset, mean, variance, FusedBatchNorm::Attrs{}.IsTraining(true)); in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | remapper_test.cc | 53 attr = attr.IsTraining(false); in TEST_F()
89 attr = attr.IsTraining(false); in TEST_F()
152 ops::FusedBatchNormV3::IsTraining(is_training) in TEST_F()
263 ops::FusedBatchNormV3::IsTraining(is_training) in TEST_F()
774 attrs = attrs.IsTraining(false); in TEST_F()
853 attrs = attrs.IsTraining(false); in TEST_F()
|
| D | generic_layout_optimizer_transposer.h | 293 bool IsTraining(const utils::MutableNodeView& node) const;
|
| D | mkl_remapper_test.cc | 338 ops::FusedBatchNormV3::IsTraining(is_training) in TEST_F()
|
| D | layout_optimizer.cc | 1220 return NodeProcessor::ShouldProcess() && IsTraining(); in ShouldProcess()
1226 bool IsTraining() const { in IsTraining() function in tensorflow::grappler::__anon5a8ea6d90111::FusedBatchNormGradProcessor
|
| D | generic_layout_optimizer_transposer.cc | 960 bool FusedBatchNormGradTransposer::IsTraining( in IsTraining() function in tensorflow::grappler::FusedBatchNormGradTransposer
977 if (!ShouldProcess(*context, *node) || !IsTraining(*node)) { in TransposeNode()
|
| D | generic_layout_optimizer_transposer_test.cc | 142 mean, var, ops::FusedBatchNormV2::IsTraining(false).Epsilon(0.1f)); in CreateSimpleFusedBatchNorm()
272 .IsTraining(is_training) in CreateSimpleFusedBatchNormGrad()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_fused_batch_norm_op_test.cc | 245 attr = attr.IsTraining(is_training); in VerifyFusedBatchNorm()
376 bn_attr = bn_attr.IsTraining(true); in VerifyFusedBatchNormGradWithConv2D()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | fused_batch_norm_ex_op_test.cc | 164 ops::FusedBatchNormV3::IsTraining(is_training) in RunFusedBatchNorm()
198 ops::FusedBatchNormGradV3::IsTraining(is_training) in RunFusedBatchNorm()
|
| D | conv_ops_test.cc | 682 attr = attr.IsTraining(false); in RunConv2DWithBatchNorm()
712 attr = attr.IsTraining(false); in RunConv2DWithBatchNormAndActivation()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 888 ops::FusedBatchNorm::IsTraining(true).Epsilon(0.1f)); in CreateGrapplerItemWithBatchNorm()
2225 ops::FusedBatchNorm::IsTraining(true).Epsilon(0.1f)); in CreateGrapplerItemWithInterDeviceTransfers()
|
| D | op_level_cost_estimator.cc | 154 bool IsTraining(const OpInfo& op_info) { in IsTraining() function
2378 const bool is_training = IsTraining(op_info); in PredictFusedBatchNorm()
|