- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for GetSparseOperands (0.64 sec)
-
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) -
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) -
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)