Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for GetSparseOperands (0.64 sec)

  1. tensorflow/compiler/mlir/lite/ir/tfl_op_interfaces.td

      let description = [{
        Interface for ops that support sparse computation.
      }];
    
      let methods = [
        InterfaceMethod<
          [{Returns the indices of sparse operands.}],
          "std::vector<int>", "GetSparseOperands", (ins)
        >,
        InterfaceMethod<
          [{Returns the supported block size of float sparse operands.}],
          "std::vector<std::vector<int>>", "GetFloatBlockSize", (ins)
        >,
        InterfaceMethod<
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 4.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc

    };
    
    void DenseToSparsePass::runOnOperation() {
      func::FuncOp func = getOperation();
      OpBuilder builder(func);
    
      func.walk([&](SparseOpInterface sparse_op) {
        const auto& sparse_operands = sparse_op.GetSparseOperands();
        std::vector<std::vector<int>> supported_block_size;
        for (int operand : sparse_operands) {
          auto* op = sparse_op.getOperation();
          auto value = op->getOperand(operand);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16.1K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

      let extraClassDeclaration = [{
        // AffineQuantizedOpInterface:
        int GetChannelDimIndex() { return 0; }
        int GetQuantizationDimIndex() { return 0; }
        // SparseOpInterface:
        std::vector<int> GetSparseOperands() { return {1}; }
        std::vector<std::vector<int>> GetFloatBlockSize() { return {}; }
        std::vector<std::vector<int>> GetQuantizedBlockSize() { return {}; }
        // DynamicRangeQuantizedOpInterface:
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
Back to top