Search Options

Results per page
Sort
Preferred Languages
Advance

Results 21 - 30 of 46 for TypeAttr (0.2 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/tfr/ir/tfr_ops.cc

        auto c_false =
            rewriter.create<arith::ConstantOp>(loc, rewriter.getBoolAttr(false));
        TypeAttr f32_attr = TypeAttr::get(rewriter.getF32Type());
        TFRAttrType output_type = TFRAttrType::get(rewriter.getContext());
        auto constant_f32_op = rewriter.create<ConstOp>(loc, output_type, f32_attr);
        TypeAttr i32_attr = TypeAttr::get(rewriter.getI32Type());
        auto constant_i32_op = rewriter.create<ConstOp>(loc, output_type, i32_attr);
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Nov 21 16:55:41 UTC 2023
    - 38.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/utils/export_utils.cc

      if (tensorflow::IsRefType(dtype)) dtype = tensorflow::RemoveRefType(dtype);
      value->set_type(dtype);
      return absl::OkStatus();
    }
    
    Status ConvertAttribute(const mlir::TypeAttr& type, bool remove_ref_type,
                            AttrValue* value) {
      return ConvertAttribute(type.getValue(), remove_ref_type, value);
    }
    
    Status ConvertAttribute(const mlir::UnitAttr& attr, AttrValue* value) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tfr/ir/tfr_ops.td

    def TFR_allowedArgType : Type<Or<[
        TFR_AllTensorTypes.predicate,
        TFR_AllAttrTypes.predicate]>, "allowed tfr.call operand types">;
    
    def TFR_allowedConstValues : Attr<Or<[
        FlatSymbolRefAttr.predicate,
        TypeAttr.predicate,
        StrAttr.predicate,
        ArrayAttr.predicate]>, "allowed tfr.constant value"> {
      let storageType = "Attribute";
      let returnType = "Attribute";
      let convertFromStorage = "$_self";
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 10:54:29 UTC 2024
    - 17.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/decompose_hybrid_quantization.cc

          Type resultTy = op->getOpResult(i).getType();
          if (QuantizedType::getQuantizedElementType(resultTy)) {
            replaceResults.push_back(rewriter.create<TFL::QuantizeOp>(
                loc, resultTy, result, TypeAttr::get(resultTy)));
            continue;
          }
    
          replaceResults.push_back(result);
        }
    
        rewriter.replaceOp(op, replaceResults);
    
        return success();
      }
    };
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tfr/passes/decompose.cc

              }
              attribute = TypeAttr::get(type);
            }
            Value attr_cst;
            // Wrap these special attributes as a special TFR constant, so the SSA
            // value has a valid type to be used as TFR function argument. These
            // attributes are not expected to be manipulated by the lowering passes.
            if (mlir::isa<TypeAttr>(attribute) || mlir::isa<ArrayAttr>(attribute) ||
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.6K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/transforms/post_quantize.cc

            output_shape,
            mlir::cast<quant::UniformQuantizedType>(output_type.getElementType())
                .getStorageType());
        rewriter.replaceOpWithNewOp<QConstOp>(
            op, TypeAttr::get(result_type),
            DenseIntElementsAttr::get(values_type, new_values));
        return success();
      }
    };
    
    // Fold constant quantized Reshape ops.
    struct FoldReshapeOp : public OpRewritePattern<ReshapeOp> {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.1K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/flatbuffer_import.cc

            "effective_hidden_scale_intermediate"};
        for (auto type_and_name :
             llvm::zip(intermediate_types, kIntermediateNames)) {
          mlir::TypeAttr type_attr =
              mlir::TypeAttr::get(std::get<0>(type_and_name));
          auto named_attr =
              builder.getNamedAttr(std::get<1>(type_and_name), type_attr);
          op_state.addAttribute(named_attr.getName(), named_attr.getValue());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 21 18:21:50 UTC 2024
    - 66.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform.cc

          if (IsQI8Type(result_type) || IsQUI8Type(result_type)) {
            builder->setInsertionPoint(op);
            TFL::QuantizeOp quant_op = builder->create<TFL::QuantizeOp>(
                op->getLoc(), result_type, new_result, TypeAttr::get(result_type));
            new_result = quant_op.getResult();
          }
    
          // Rewire the outputs.
          result.replaceAllUsesWith(new_result);
        }
    
        // Remove the old op.
        op->erase();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h

    // the casting, the quantization dimension of the result type needs to be set
    // this new `axis` value.
    TypeAttr CastQuantizedTypeAttrFromExpressedType(Builder builder,
                                                    TypeAttr source, Type target,
                                                    int axis);
    
    // Quantizes the elements in the attribute `real_value` by the quantization
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Apr 24 20:30:06 UTC 2024
    - 41.7K bytes
    - Viewed (0)
Back to top