Home
last modified time | relevance | path

Searched refs:kFusedBatchNorm (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops_fused_impl.h80 kFusedBatchNorm, // CPU only enumerator
345 case FusedComputationType::kFusedBatchNorm:
861 fused_computation_ = FusedComputationType::kFusedBatchNorm;
881 if (fused_computation_ == FusedComputationType::kFusedBatchNorm ||
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc66 constexpr char kFusedBatchNorm[] = "FusedBatchNorm"; variable
279 {kFusedBatchNorm, wrap(&OpLevelCostEstimator::PredictFusedBatchNorm)}, in OpLevelCostEstimator()