Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for dynamic_slice (0.2 sec)

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

          // Dynamic start indices should be clamped with upper limit of
          // `shape(operand) - slice_sizes)` as per semantics of
          // `stablehlo.dynamic_slice`.
          // (https://github.com/openxla/stablehlo/blob/main/docs/spec.md#dynamic_slice)
          start_indices[i] =
              rewriter.create<TFL::MinimumOp>(op->getLoc(), cast, upper_limit_cst);
        }
    
        Value concatenated = start_indices[0];
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 22 09:00:19 UTC 2024
    - 99.8K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/stablehlo/tests/uniform-quantized-stablehlo-to-tfl.mlir

    // CHECK-NOT: tfl.gather
    
    // -----
    
    // Tests that a quantized `stablehlo.dynamic_slice` is converted to `tfl.slice`.
    
    // CHECK-LABEL: func @dynamic_slice
    // CHECK-SAME: %[[ARG0:.+]]: tensor<4x4x!quant.uniform<i8:f32, 3.000000e-01:-5>>, %[[ARG1:.+]]: tensor<i64>, %[[ARG2:.+]]: tensor<i64>
    func.func @dynamic_slice(
        %arg0: tensor<4x4x!quant.uniform<i8:f32, 3.000000e-01:-5>>,
        %arg1: tensor<i64>,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 14 17:10:32 UTC 2024
    - 106.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc

        Type start_indices_element_type =
            mlir::cast<ShapedType>(op.getStartIndices().front().getType())
                .getElementType();
    
        // The mhlo dynamic_slice's start_indices can be either signed/unsigned
        // int32/int64. However, TF only takes in either i32 or i64 types for begin,
        // so we will always put a cast.
        Type signed_start_indices_element_type;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 154.9K bytes
    - Viewed (0)
Back to top