Searched refs:PredictFusedBatchNorm (Results 1 – 3 of 3) sorted by relevance
146 Costs PredictFusedBatchNorm(const OpContext& op_context) const;
279 {kFusedBatchNorm, wrap(&OpLevelCostEstimator::PredictFusedBatchNorm)}, in OpLevelCostEstimator()1647 Costs OpLevelCostEstimator::PredictFusedBatchNorm( in PredictFusedBatchNorm() function in tensorflow::grappler::OpLevelCostEstimator
1175 TEST_F(OpLevelCostEstimatorTest, PredictFusedBatchNorm) { in TEST_F() argument