Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 16 of 16 for output_shapes (0.4 sec)

  1. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

          }
          auto output_shape =
              mlir::cast<RankedTensorType>(conv_op.getResult().getType())
                  .getShape();
          SmallVector<int64_t, 4> transposed_output_shape = {
              output_shape[dnums.getOutputBatchDimension()],
              output_shape[dnums.getOutputSpatialDimensions().data()[0]],
              output_shape[dnums.getOutputSpatialDimensions().data()[1]],
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/optimize.cc

          mlir::cast<ShapedType>(input.getType()).getShape();
      ArrayRef<int64_t> output_shape =
          mlir::cast<ShapedType>(output.getType()).getShape();
    
      int64_t agg_value = 1;
      for (size_t i = agg_start_idx; i < input_shape.size() - 1; ++i) {
        agg_value *= input_shape[i];
      }
    
      return (agg_value == output_shape[agg_start_idx]);
    }
    
    // Returns whether the given type `a` is broadcast-compatible with `b`.
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 00:40:15 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/quantization/tensorflow/python/integration_test/quantize_model_test.py

                        )
                    ]
                )
            )
            self.assertTrue(
                self._contains_op(
                    output_graphdef,
                    'Const',
                    '_output_shapes',
                    per_channel_size_attr,
                )
            )
        elif target_opset == quant_opts_pb2.UNIFORM_QUANTIZED:
          self.assertTrue(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 03:36:50 UTC 2024
    - 235.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc

                    .getScale(),
                /*filter_scales=*/filter_quantized_type.getScales());
            const ArrayRef<int64_t> output_shape =
                op->getResult(0).getType().cast<TensorType>().getShape();
            const SmallVector<int64_t, 1> bias_shape = {
                output_shape[output_shape.size() - 1]};
            // `tfl.fully_connected`'s `GetChannelDimIndex` is 0.
            const auto bias_quantized_type =
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/ir/tfl_ops.td

      let summary = "Transpose convolution operator";
    
      let description = [{
        Performs transpose convolution operation on input.
      }];
    
      let arguments = (ins
        TFL_I32Tensor:$output_shape,
        TFL_TensorOf<[F32, QI8, QUI8, QI16]>:$weights,
        TFL_TensorOf<[F32, QI8, QUI8, QI16]>:$input,
        TFL_TensorOfOrNone<[F32, QI32, I64]>:$bias,
        TFL_PaddingAttr:$padding,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Jun 06 19:09:08 UTC 2024
    - 186K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

                        IsWithinInt32Range);
    
        if (elements_all_in_int32_range) {
          std::vector<int32_t> output_shape(output_ty.getRank());
          std::transform(output_ty.getShape().begin(), output_ty.getShape().end(),
                         output_shape.begin(),
                         [](int64_t val) { return static_cast<int32_t>(val); });
          output_int_type = tensorflow::GetTypeFromTFTensorShape(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
Back to top