Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for SparsityParameterAttr (0.3 sec)

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

      }
    
      return result;
    }
    
    template <typename T>
    std::vector<T> BuildSparsityParameterAttribute(
        const std::vector<int>& block_size, const T* dense_buffer, Operation* inst,
        OpBuilder* builder, SparsityParameterAttr* s_param) {
      ElementsAttr attr;
      ShapedType type;
      if (auto cst = dyn_cast<ConstOp>(inst)) {
        attr = cst.getValue();
        type = mlir::cast<ShapedType>(cst.getType());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 16.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/ir/tfl_op_enums.td

          "int32_t":$dense_size,
          I32ArrayParameter:$segments,
          I32ArrayParameter:$indices
      );
      let summary = "Dimension metadata.";
      let assemblyFormat = "`<` struct(params) `>`";
    }
    
    def SparsityParameterAttr : AttrDef<TFL_Dialect, "SparsityParameter"> {
      let mnemonic = "sparsity_parameter";
      let parameters = (ins
          I32ArrayParameter:$traversal_order,
          I32ArrayParameter:$block_map,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 20 00:05:24 UTC 2022
    - 6.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

      }];
    
      let arguments = (ins ElementsAttr:$value,
                       SparsityParameterAttr:$s_param,
                       ElementsAttr:$compressed_data);
    
      let results = (outs AnyTensor:$output);
    
      let builders = [
        OpBuilder<(ins "TypedAttr":$value, "SparsityParameterAttr":$s_param,
          "Attribute":$compressed_data),
        [{
          $_state.addTypes(value.getType());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

              0, segments, indices);
        } else {
          return errors::Unimplemented("Unsupported dimension metadata type");
        }
      }
      auto s_param = tfl::SparsityParameterAttr::get(
          builder.getContext(), tensor.sparsity->traversal_order,
          tensor.sparsity->block_map, dim_metadata);
    
      std::vector<char> dense_buffer(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

      // Returns a unique name for `val`.
      std::string UniqueName(mlir::Value val);
    
      BufferOffset<tflite::SparsityParameters> BuildSparsityParameters(
          const mlir::TFL::SparsityParameterAttr& s_attr);
    
      bool EstimateArithmeticCount(int64_t* count);
    
      // Check compatibility with GPU delegate and returns the compatibility.
      bool CheckGpuDelegateCompatibility(uint8_t* model_buffer_pointer);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
Back to top