Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for input_tensor (0.23 sec)

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

            ]
        )
        def __call__(self, input_tensor: core.Tensor) -> Mapping[str, core.Tensor]:
          """Performs a matrix multiplication.
    
          Args:
            input_tensor: Input tensor to matmul with the filter.
    
          Returns:
            A map of: output key -> output result.
          """
    
          out = math_ops.matmul(input_tensor, self.filters)
          return {'output': out}
    
    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/ir/tfl_ops.cc

      if (input_tensor.isSplat()) {
        return input_tensor.reshape(input_tensor.getType().cloneWith(
            output_shape, input_tensor.getElementType()));
      }
    
      // MLIR implementation pads elements < 8 bits to 8 bits and pads non byte
      // aligned to the nearest byte. So this is allowed.
      const char* raw_input = input_tensor.getRawData().data();
      const int element_byte_size =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 169.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

    // "third_party/tensorflow/compiler/xla/xla_data.pb.h" into
    // "third_party/tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc" is
    // resolved
    LogicalResult PrecheckForXlaConvV2Op(XlaConvV2Op op) {
      auto input_tensor = op.getLhs();
      auto kernel_tensor = op.getRhs();
      auto window_strides = op.getWindowStrides();
      auto padding = op.getPadding();
      auto lhs_dilation = op.getLhsDilation();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

        QuantizableResult,
        Pure]> {
      let summary = "Mean operator";
    
      let description = [{
        Computes the mean of elements across dimensions of a tensor.
        Reduces input_tensor along the dimensions given in axis.
        Unless keepdims is true, the rank of the tensor is reduced by 1 for
        each entry in axis. If keepdims is true, the reduced dimensions are retained
        with length 1.
      }];
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
Back to top