- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for SparseOpInterface (0.15 sec)
-
tensorflow/compiler/mlir/lite/ir/tfl_op_interfaces.td
>, ]; } //===----------------------------------------------------------------------===// // TFL op interface for sparse operands. def TFL_SparseOp : OpInterface<"SparseOpInterface"> { let description = [{ Interface for ops that support sparse computation. }]; let methods = [ InterfaceMethod< [{Returns the indices of sparse operands.}],
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/quantization/tools/tflite_op_coverage_spec_getters_gen.cc
os.indent(4) << "new std::set<std::string>({\n"; // Retrieve all the ops that have SparseOp trait. for (const auto *def : defs) { Operator op(def); if (!op.getTrait("SparseOpInterface::Trait")) { continue; } os.indent(6) << "\"" << op.getCppClassName() << "\",\n"; } os.indent(4) << "});"; os.indent(2) << "return *result;\n"; os.indent(0) << "}\n"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/dense_to_sparse.cc
void runOnOperation() override; }; 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();
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 hasVerifier = 1; 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 {}; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0)