Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for InferPaddedTensorShape (0.12 sec)

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

        const auto input_tensor_type =
            input_value.getType().cast<RankedTensorType>();
        const int64_t rank = input_tensor_type.getRank();
    
        SmallVector<int64_t> padded_output_tensor_shape =
            InferPaddedTensorShape(input_tensor_type.getShape(), tfl_pad_values);
    
        auto padded_output_tensor_type = RankedTensorType::get(
            padded_output_tensor_shape, input_tensor_type.getElementType());
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
Back to top