Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 41 for new_axis_mask (0.24 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/canonicalize.mlir

      %2 = "tf.Shape"(%arg0) : (tensor<?x1x2x?xf32>) -> tensor<4xi32>
      %3 = "tf.StridedSlice"(%2, %1, %0, %1) {begin_mask = 0 : i64, ellipsis_mask = 0 : i64, end_mask = 0 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 0 : i64} : (tensor<4xi32>, tensor<1xi32>, tensor<1xi32>, tensor<1xi32>) -> tensor<2xi32>
      func.return %3 : tensor<2xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 132.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir

      func.return %0 : tensor<1x2x2x5xf32>
      // CHECK-LABEL: strided_slice
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 05 01:54:33 UTC 2024
    - 153.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/metrics/error_collector_inst_test.cc

              "Failed at tf.StridedSlice op\nsee current operation: %2 = "
              "\"tf.StridedSlice\"(%arg0, %1, %1, %0) <{begin_mask = 11 : "
              "i64, ellipsis_mask = 0 : i64, end_mask = 11 : i64, new_axis_mask = "
              "4 : i64, shrink_axis_mask = 0 : i64}> {device = \"\"} : "
              "(tensor<*xf32>, tensor<4xi32>, tensor<4xi32>, tensor<4xi32>) "
              "-> tensor<*xf32>\nError code: ERROR_NEEDS_FLEX_OPS",
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Mar 25 01:48:36 UTC 2024
    - 8.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

        if (ellipsis_seen && IsSet(sparse.new_axis_mask, sparse_index))
          num_new_axis_after_ellipsis++;
        if (IsSet(sparse.ellipsis_mask, sparse_index)) ellipsis_seen = true;
      }
    
      int dense_index = 0;
      for (int sparse_index = 0; sparse_index < sparse.dims; ++sparse_index) {
        if (IsSet(sparse.new_axis_mask, sparse_index)) continue;
        if (IsSet(sparse.ellipsis_mask, sparse_index)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/transforms/legalize_patterns.td

        $ellipsis_mask, $new_axis_mask, $shrink_axis_mask),
      (TFL_StridedSliceOp $input,
        (CreateTFCastToInt32Op $begin), (CreateTFCastToInt32Op $end),
        (CreateTFCastToInt32Op $strides), (convertIntAttrTo32Bit $begin_mask),
        (convertIntAttrTo32Bit $end_mask), (convertIntAttrTo32Bit $ellipsis_mask),
        (convertIntAttrTo32Bit $new_axis_mask),
        (convertIntAttrTo32Bit $shrink_axis_mask),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 04 13:30:42 UTC 2024
    - 28.5K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc

          /*begin_mask=*/builder.getIntegerAttr(attribute_type, 0),
          /*end_mask=*/builder.getIntegerAttr(attribute_type, 0),
          /*ellipsis_mask=*/builder.getIntegerAttr(attribute_type, 0),
          /*new_axis_mask=*/builder.getIntegerAttr(attribute_type, 0),
          /*shrink_axis_mask=*/builder.getIntegerAttr(attribute_type, 1));
    }
    
    // Given Value input_size, and known numbers filter_sz, dilation_rate, stride,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 13.3K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/tests/optimize.mlir

      // CHECK:  %1 = "tfl.strided_slice"(%arg0, %arg1, %[[cst]], %[[c0]]) <{begin_mask = 0 : i32, ellipsis_mask = 0 : i32, end_mask = 0 : i32, new_axis_mask = 0 : i32, offset = true, shrink_axis_mask = 0 : i32}> : (tensor<4xi32>, tensor<1xi32>, tensor<1xi32>, tensor<1xi32>) -> tensor<4xi32>
    
      %cst_0 = "tfl.pseudo_const"() {value = dense<1> : tensor<i32>} : () -> tensor<i32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 20:31:41 UTC 2024
    - 284.1K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/lite/tests/prepare-composite-functions-tf.mlir

    // CHECK:           [[VAL_24:%.*]] = "tf.StridedSlice"([[VAL_20]], [[VAL_21]], [[VAL_22]], [[VAL_23]]) <{begin_mask = 6 : i64, ellipsis_mask = 0 : i64, end_mask = 6 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 1 : i64}> : (tensor<?x8x10xf32>, tensor<3xi32>, tensor<3xi32>, tensor<3xi32>) -> tensor<8x10xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 122.1K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/tests/const-fold.mlir

      %2 = "tfl.pseudo_const"() {value = dense<1> : tensor<1xi32>} : () -> tensor<1xi32>
      %3 = "tfl.strided_slice"(%arg0, %1, %0, %2) {begin_mask = 0 : i32, ellipsis_mask = 0 : i32, end_mask = 0 : i32, new_axis_mask = 0 : i32, shrink_axis_mask = 0 : i32, offset = false} : (tensor<15600xf32>, tensor<1xi32>, tensor<1xi32>, tensor<1xi32>) -> tensor<15600xf32>
      func.return %3 : tensor<15600xf32>
      // CHECK:  return %arg0
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 45.8K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-prefer-tf2xla.mlir

      %strides = "tf.Const"() {value = dense<[1, 3]> : tensor<2xi32>} : () -> (tensor<2xi32>)
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 06 15:32:52 UTC 2024
    - 15.8K bytes
    - Viewed (0)
Back to top