Search Options

Results per page
Sort
Preferred Languages
Advance

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

  1. tensorflow/compiler/mlir/lite/quantization/lite/quantize_model_test.cc

      ASSERT_THAT(weights_zero_points, SizeIs(out_channel_size));
      ASSERT_THAT(input_tensor->quantization->scale, SizeIs(1));
      ASSERT_THAT(output_tensor->quantization->scale, SizeIs(1));
    
      const float eps = 1e-7;
    
      // Bias scale should be input * per_channel_weight_scale.
      for (size_t i = 0; i < out_channel_size; i++) {
        EXPECT_THAT(bias_scales[i], FloatNear(input_tensor->quantization->scale[0] *
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 23:15:24 UTC 2024
    - 73.9K bytes
    - Viewed (0)
  2. 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)
  3. 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)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      # 2. invert(a) or a = invert(0)
      input_tensor = tf.constant([0, 5, 3, 14], dtype=dtype)
      not_a_and_a, not_a_or_a, not_0 = [bitwise_ops.bitwise_and(
                                          input_tensor, bitwise_ops.invert(input_tensor)),
                                        bitwise_ops.bitwise_or(
                                          input_tensor, bitwise_ops.invert(input_tensor)),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top