Search Options

Results per page
Sort
Preferred Languages
Advance

Results 51 - 59 of 59 for i32 (0.04 sec)

  1. tensorflow/compiler/mlir/quantization/common/uniform_quantized_types.h

                                                         bool narrow_range = false);
    
    // Creates a `UniformQuantizedType` with the given `scale` and `zero_point`
    // values. The produced type has f32 as its expressed type and i32 as its
    // storage type. The available values use the full range of the storage value.
    // Assumes asymmetric quantization, meaning the zero point value can be
    // a non-zero value.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/common/uniform_quantized_types.cc

        LLVM_DEBUG(llvm::dbgs()
                   << "Expected a uniform quantized type. Got: " << type << ".\n");
        return false;
      }
    
      if (!IsStorageTypeI32(quantized_type)) {
        LLVM_DEBUG(llvm::dbgs() << "Expected an i32 storage type. Got: "
                                << quantized_type << ".\n");
        return false;
      }
    
      if (!IsExpressedTypeF32(quantized_type)) {
        LLVM_DEBUG(llvm::dbgs() << "Expected an f32 expressed type. Got: "
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  3. src/cmd/compile/internal/test/mergelocals_test.go

    	"cmd/compile/internal/types"
    	"cmd/internal/src"
    	"internal/testenv"
    	"path/filepath"
    	"sort"
    	"strings"
    	"testing"
    )
    
    func mkiv(name string) *ir.Name {
    	i32 := types.Types[types.TINT32]
    	s := typecheck.Lookup(name)
    	v := ir.NewNameAt(src.NoXPos, s, i32)
    	return v
    }
    
    func TestMergeLocalState(t *testing.T) {
    	v1 := mkiv("v1")
    	v2 := mkiv("v2")
    	v3 := mkiv("v3")
    
    	testcases := []struct {
    Registered: Wed Jun 12 16:32:35 UTC 2024
    - Last Modified: Thu Apr 18 15:43:53 UTC 2024
    - 4.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tfrt/tests/batch_function_lowering.mlir

      func.return %1 : tensor<*xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 08 00:18:59 UTC 2024
    - 2K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/experimental/tac/transforms/device_transform.cc

      });
    }
    
    // Fold quantized i32 (normally bias) into their float values.
    struct FoldQuantizedI32ToFloat : public OpRewritePattern<TFL::DequantizeOp> {
      using OpRewritePattern<TFL::DequantizeOp>::OpRewritePattern;
    
      LogicalResult matchAndRewrite(TFL::DequantizeOp dequant_op,
                                    PatternRewriter& rewriter) const override {
        // We only fold i32 -> float pattern.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config_test.cc

    namespace mlir {
    namespace mhlo {
    
    using func::FuncOp;
    using mlir::ModuleOp;
    
    static constexpr char kMlirModuleStr[] = R"(
    module attributes {tf.versions = {bad_consumers = [], min_consumer = 0 : i32, producer = 1442 : i32}} {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 30 03:31:01 UTC 2024
    - 8.4K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/quantization/stablehlo/passes/convert_func_to_bfloat16.cc

        });
      }
    };
    
    // This helper function makes legality check easier. Both convert ops in the
    // patterns below are considered legal:
    //  - `BitcastConvertOp` (i32 -> f32) + `ConvertOp` (f32 -> bf16)
    //  - `ConvertOp` (bf16 -> f32) -> `BitcastConvertOp` (f32 -> i32)
    template <typename ConvertOp, typename OtherConvertOp>
    bool IsConvertOpLegal(ConvertOp convert_op, BFloat16TypeConverter& converter) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.3K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_stablehlo_composite_to_tfl_custom.cc

    // composites. See `IsSupportedComposite` for list of supported ops.
    //
    // Example:
    //   %0 = stablehlo.composite "odml.some_op" <args> {
    //      composite_attrs = {<attrs>},
    //      version = 0 : i32
    //   }
    //   ==>
    //   %0 = tfl.custom(<args>) {
    //     custom_code = "odml.some_op",
    //     custom_option = #tfl<const_bytes : "flexbuffer_serialized_attrs">
    //   }
    struct LegalizeCompositeToCustomOpPass
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 6.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/optimize_batch_matmul.cc

          RankedTensorType input_type =
              mlir::cast<RankedTensorType>(input.getType());
          const int input_rank = input_type.getRank();
    
          // Create a 1D I32 tensor for representing the dimension permutation.
          auto permuation_tensor_type =
              RankedTensorType::get({input_rank}, rewriter.getIntegerType(32));
          llvm::SmallVector<Attribute, 4> permute;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 9.6K bytes
    - Viewed (0)
Back to top