diff options
Diffstat (limited to 'tensorflow/core/util/sparse/dim_comparator.h')
| -rw-r--r-- | tensorflow/core/util/sparse/dim_comparator.h | 16 |
1 files changed, 8 insertions, 8 deletions
diff --git a/tensorflow/core/util/sparse/dim_comparator.h b/tensorflow/core/util/sparse/dim_comparator.h index b773b33008..0782e7e1a8 100644 --- a/tensorflow/core/util/sparse/dim_comparator.h +++ b/tensorflow/core/util/sparse/dim_comparator.h @@ -13,8 +13,8 @@ See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ -#ifndef TENSORFLOW_UTIL_SPARSE_DIM_COMPARATOR_H_ -#define TENSORFLOW_UTIL_SPARSE_DIM_COMPARATOR_H_ +#ifndef TENSORFLOW_CORE_UTIL_SPARSE_DIM_COMPARATOR_H_ +#define TENSORFLOW_CORE_UTIL_SPARSE_DIM_COMPARATOR_H_ #include "third_party/eigen3/unsupported/Eigen/CXX11/Tensor" #include "tensorflow/core/kernels/bounds_check.h" @@ -49,11 +49,11 @@ class DimComparator { DimComparator(const TTypes<int64>::Matrix& ix, const VarDimArray& order, const VarDimArray& shape) : ix_(ix), order_(order), dims_(shape.size()) { - CHECK_GT(order.size(), size_t{0}) << "Must order using at least one index"; - CHECK_LE(order.size(), shape.size()) << "Can only sort up to dims"; + DCHECK_GT(order.size(), size_t{0}) << "Must order using at least one index"; + DCHECK_LE(order.size(), shape.size()) << "Can only sort up to dims"; for (size_t d = 0; d < order.size(); ++d) { - CHECK_GE(order[d], 0); - CHECK_LT(order[d], shape.size()); + DCHECK_GE(order[d], 0); + DCHECK_LT(order[d], shape.size()); } } @@ -97,7 +97,7 @@ class FixedDimComparator : DimComparator { FixedDimComparator(const TTypes<int64>::Matrix& ix, const VarDimArray& order, const VarDimArray& shape) : DimComparator(ix, order, shape) { - CHECK_EQ(order.size(), ORDER_DIM); + DCHECK_EQ(order.size(), ORDER_DIM); } inline bool operator()(const int64 i, const int64 j) const { bool value = false; @@ -116,4 +116,4 @@ class FixedDimComparator : DimComparator { } // namespace sparse } // namespace tensorflow -#endif // TENSORFLOW_UTIL_SPARSE_DIM_COMPARATOR_H_ +#endif // TENSORFLOW_CORE_UTIL_SPARSE_DIM_COMPARATOR_H_ |