Searched refs:IsTraining (Results 1 – 6 of 6) sorted by relevance
102 scale, offset, mean, variance, FusedBatchNorm::Attrs{}.IsTraining(true)); in TEST()105 scale, offset, mean, variance, FusedBatchNorm::Attrs{}.IsTraining(true)); in TEST()
48 attr = attr.IsTraining(false); in TEST_F()81 attr = attr.IsTraining(false); in TEST_F()253 attrs = attrs.IsTraining(false); in TEST_F()330 attrs = attrs.IsTraining(false); in TEST_F()
1161 return NodeProcessor::ShouldProcess() && IsTraining(); in ShouldProcess()1167 bool IsTraining() const { in IsTraining() function in tensorflow::grappler::__anonbd4a17fb0111::FusedBatchNormGradProcessor
664 attr = attr.IsTraining(false); in RunConv2DWithBatchNorm()691 attr = attr.IsTraining(false); in RunConv2DWithBatchNormAndRelu()
106 bool IsTraining(const OpInfo& op_info) { in IsTraining() function1658 const bool is_training = IsTraining(op_info); in PredictFusedBatchNorm()
271 ops::FusedBatchNorm::IsTraining(true).Epsilon(0.1f)); in CreateGrapplerItemWithBatchNorm()1559 ops::FusedBatchNorm::IsTraining(true).Epsilon(0.1f)); in CreateGrapplerItemWithInterDeviceTransfers()