Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for bias_shape (0.53 sec)

  1. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py

                num_bits=8,
                narrow_range=False,
            )
            return {'output': out}
    
        bias = None
        if has_bias:
          bias_shape = shapes[1][-1]
          if bias_shape is not None:
            bias = array_ops.constant(
                np.random.uniform(size=[shapes[1][-1]]), dtype=dtypes.float32
            )
        model = MatmulModel(bias)
        x = array_ops.constant(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 03:36:50 UTC 2024
    - 235.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

            filter_quantized_element_type.getZeroPoint());
      }
    
      SmallVector<int64_t, 1> bias_shape = {filter_shape[0]};
      auto bias_type =
          RankedTensorType::getChecked(loc, bias_shape, bias_quantized_type);
    
      auto bias_value_type = RankedTensorType::getChecked(
          loc, std::move(bias_shape), rewriter.getI32Type());
      auto bias_value = DenseIntElementsAttr::get(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

    }
    
    std::optional<RankedTensorType> InferWindowOutputShape(
        const ShapedType& base_shape, const xla::Window& window,
        Type element_type) {
      if (window.dimensions_size() != base_shape.getRank()) {
        llvm::errs() << "Window has dimension " << window.dimensions_size()
                     << " but base shape has dimension " << base_shape.getRank()
                     << "\n";
        return std::nullopt;
      }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
Back to top