Searched refs:elements_shape (Results 1 – 3 of 3) sorted by relevance
108 bool IsDimensionsDegenerateExceptLastOne(ArrayRef<int64_t> elements_shape) { in IsDimensionsDegenerateExceptLastOne() argument109 if (elements_shape.empty()) return true; in IsDimensionsDegenerateExceptLastOne()111 for (auto dim : elements_shape.drop_back(1)) { in IsDimensionsDegenerateExceptLastOne()
110 bool IsDimensionsDegenerateExceptLastOne(ArrayRef<int64_t> elements_shape);
121 const ArrayRef<int64_t> elements_shape, in CanFuseConvOrDepthwiseConvShapes() argument126 const auto elements_rank = elements_shape.size(); in CanFuseConvOrDepthwiseConvShapes()127 for (int i = 0; i < static_cast<int>(elements_shape.size()) - 1; ++i) { in CanFuseConvOrDepthwiseConvShapes()128 if (elements_shape[i] != 1) return false; in CanFuseConvOrDepthwiseConvShapes()133 auto elements_depth = elements_shape.empty() ? 1 : elements_shape.back(); in CanFuseConvOrDepthwiseConvShapes()158 const auto elements_shape = elements.getType().getShape(); in CanFuseConvOrDepthwiseConv() local160 return CanFuseConvOrDepthwiseConvShapes(filter_shape, elements_shape, in CanFuseConvOrDepthwiseConv()