Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dfused_eigen_output_kernels.cc76 if (*fused_computation == FusedComputationType::kFusedBatchNorm || in InitializeFusedComputation()
Dfused_eigen_output_kernels.h43 kFusedBatchNorm, enumerator
160 return fusion == FusedComputationType::kFusedBatchNorm || in IsSupported()
Dconv_ops_fused_impl.h274 case FusedComputationType::kFusedBatchNorm:
736 {FCT::kFusedBatchNorm, {"FusedBatchNorm"}},
/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc98 constexpr char kFusedBatchNorm[] = "FusedBatchNorm"; variable
517 device_cost_impl_.emplace(kFusedBatchNorm, in OpLevelCostEstimator()