diff options
author | Yao Zhang <yaozhang@google.com> | 2018-06-20 18:36:13 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-06-20 18:39:22 -0700 |
commit | 96dfcc2fdc9f3a7419d3d5c5a64489e757de624e (patch) | |
tree | 8c684731bde1643158037bf1d4ed17e58c95096a /tensorflow/core/grappler/costs/op_level_cost_estimator.h | |
parent | e8b18a6f0c02d364ff47ba5fa3dc61458d273674 (diff) |
Support filter format for FusedConv2DBiasActivation.
PiperOrigin-RevId: 201454730
Diffstat (limited to 'tensorflow/core/grappler/costs/op_level_cost_estimator.h')
-rw-r--r-- | tensorflow/core/grappler/costs/op_level_cost_estimator.h | 10 |
1 files changed, 0 insertions, 10 deletions
diff --git a/tensorflow/core/grappler/costs/op_level_cost_estimator.h b/tensorflow/core/grappler/costs/op_level_cost_estimator.h index d384f57279..a277dfdf65 100644 --- a/tensorflow/core/grappler/costs/op_level_cost_estimator.h +++ b/tensorflow/core/grappler/costs/op_level_cost_estimator.h @@ -84,13 +84,6 @@ class OpLevelCostEstimator { int64 sy; // Stride y. Padding padding; // SAME or VALID. }; - enum ConvolutionFormat { - UNKNOWN_CONVOLUTION_FORMAT, - NHWC, - NCHW, - NCHW_VECT_C, - NCHW_VECT_W, - }; int64 CountConv2DOperations(const OpInfo& op_features, bool* found_unknown_shapes) const; int64 CountConv2DOperations(const OpInfo& op_features, @@ -198,9 +191,6 @@ class OpLevelCostEstimator { static OpInfo::TensorProperties DescribeTensor( DataType type, const std::vector<int64>& dims); - // Returns the Conv2D format for this operation. - static ConvolutionFormat GetConvolutionFormat(const OpContext& op_context); - // This method calculates the execution time depending on whether IO can // overlap with computation. It assumes the memory and the compute times have // already been calculated. |