Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for sizeclass (0.15 sec)

  1. tensorflow/compiler/mlir/lite/transforms/optimize.cc

        int input_last_dim = input_ty.getShape().back();
        if (input_last_dim < 1) return std::nullopt;
        int num_result = result_ty.getNumElements();
        auto size_last = *(--size.value_end<APInt>());
        int32_t k = size_last.getSExtValue();
        if (num_input / input_last_dim * k != num_result) return std::nullopt;
        // We don't match sliceOp with last dim size = 0.
        if (!k) return std::nullopt;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 102.3K bytes
    - Viewed (0)
Back to top