Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for TypeAttr (0.68 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

      auto new_filter_constant_value_attr =
          DenseIntElementsAttr::get(new_filter_value_attr_type, new_filter_values);
      return rewriter.create<TFL::QConstOp>(
          filter_constant_op.getLoc(),
          /*output=*/TypeAttr::get(new_filter_result_type),
          /*value=*/new_filter_constant_value_attr);
    }
    
    // Creates the `tfl.qconst` for filter. If `rhs_op` is a `stablehlo.constant`,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        // determined while going through quantization passes.
        OptionalAttr<TypeAttr>:$input_to_input_intermediate,
        OptionalAttr<TypeAttr>:$input_to_forget_intermediate,
        OptionalAttr<TypeAttr>:$input_to_cell_intermediate,
        OptionalAttr<TypeAttr>:$input_to_output_intermediate,
        OptionalAttr<TypeAttr>:$effective_hidden_scale_intermediate
      );
    
      let results = (outs AnyTensor:$output);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/optimize.cc

    }
    
    ElementsAttr ExpandTo4DForConv(Attribute a) {
      return ExpandTo4DForConvImpl(a, false);
    }
    
    ElementsAttr ExpandTo4DForDepthwiseConv(Attribute a) {
      return ExpandTo4DForConvImpl(a, true);
    }
    
    TypeAttr RescaleQtype(Type input, Attribute factor) {
      return quant::RescaleQuantizedType(input, factor);
    }
    
    // Returns `true` if reducing `axes` in `input` with `keep_dims=true` results in
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/translate/import_model.cc

                builder.getUnknownLoc(),
                builder.getStringAttr(object_names.GetSymbolTableName(node_id)),
                mlir::ElementsAttr(),
                /*type=*/mlir::TypeAttr::get(type),
                /*is_mutable=*/builder.getUnitAttr());
            op->setAttr(
                kTfSavedModelExportedNamesAttr,
                builder.getStrArrayAttr(object_names.GetExportedNames(node_id)));
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 01 11:17:36 UTC 2024
    - 183.2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

    // attributes.
    Type GetType(Attribute shape_attr, Attribute type_attr) {
      auto shape = mlir::cast<tf_type::ShapeAttr>(shape_attr);
      auto type = mlir::cast<TypeAttr>(type_attr);
      if (shape.hasRank())
        return tensorflow::GetTypeFromTFTensorShape(shape.getShape(),
                                                    type.getValue());
      else
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

          for (const std::string& intermediate : intermediate_names) {
            auto intermediate_attr = inst.getAttr(intermediate);
            if (auto attr =
                    mlir::dyn_cast_or_null<mlir::TypeAttr>(intermediate_attr)) {
              Type qtype = attr.getValue();
              auto tensor_or = BuildTensorFromType(
                  qtype, name_mapper_.GetUniqueName(intermediate).str());
              if (!tensor_or.has_value()) {
    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