- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for NextTensorIndex (0.15 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/hlo_matchers.cc
// If there is no valid next index, returns `std::nullopt`. // // `index` should have the same size as `shape`. // Each value `dim` in `index` should be in [0, shape[dim]). static std::optional<SmallVector<int64_t>> NextTensorIndex( SmallVector<int64_t> index, ArrayRef<int64_t> shape, int64_t fixed_axis) { #ifndef NDEBUG assert(shape.size() == index.size()); assert(fixed_axis < shape.size()); assert(fixed_axis >= 0);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.6K bytes - Viewed (0)