Search Options

Results per page
Sort
Preferred Languages
Advance

Results 11 - 20 of 39 for ellipsis_mask (0.32 sec)

  1. tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc

        uint64_t ellipsis_mask = strided_slice_op.getEllipsisMask();
        uint64_t shrink_axis_mask = strided_slice_op.getShrinkAxisMask();
        uint64_t new_axis_mask = strided_slice_op.getNewAxisMask();
    
        // Enforce operator precedence.
        shrink_axis_mask &= ~ellipsis_mask;
        new_axis_mask &= ~ellipsis_mask;
    
        DenseIntElementsAttr begin_dense_elem_attr;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 28 21:49:50 UTC 2024
    - 64.6K 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

              "MockFailurePass",
              "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>) "
    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/lite/transforms/legalize_patterns.td

        $input, $begin, $end, $strides, $begin_mask, $end_mask,
        $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),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 04 13:30:42 UTC 2024
    - 28.5K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_xla_attribute_utils.cc

          /*strides=*/Create1DConstValue<int32_t>(builder, loc, {1}),
          /*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));
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 13.3K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/lite/tests/optimize.mlir

      // CHECK-DAG:  %[[c0:.*]] = "tfl.pseudo_const"() <{value = dense<1> : tensor<1xi32>}> : () -> tensor<1xi32>
      // 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>
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 16 20:31:41 UTC 2024
    - 284.1K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/tests/prepare-composite-functions-tf.mlir

    // CHECK-DAG:       [[VAL_23:%.*]] = arith.constant dense<1> : tensor<3xi32>
    // 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)
  8. 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>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 45.8K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf-prefer-tf2xla.mlir

    %strides = "tf.Const"() {value = dense<[1, 3]> : tensor<2xi32>} : () -> (tensor<2xi32>)
    
      // expected-remark@+1 {{failed to create tf2xla kernel: INVALID_ARGUMENT: NodeDef missing attrs 'shrink_axis_mask', 'new_axis_mask', 'begin_mask', 'ellipsis_mask', 'end_mask' from Op<name=StridedSlice; signature=input:T, begin:Index, end:Index, strides:Index -> output:T; attr=T:type; attr=Index:type,allowed=[DT_INT16, DT_INT32, DT_INT64]; attr=begin_mask:int,default=0; attr=end_mask:int,default=0; attr...
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Apr 06 15:32:52 UTC 2024
    - 15.8K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir

      // CHECK-DAG-SAME: strides = dense<[1, 1, 1, 1, 1, 2]> : tensoe<6xi64>
      // CHECK-SAME: -> tensor<1x4x8x8x10x2xf32>
      %0 = "tf.StridedSlice"(%input, %begin, %end, %strides) {begin_mask = 8, end_mask = 4, shrink_axis_mask = 1, ellipsis_mask = 2} : (tensor<2x4x8x16x32x64xf32>, tensor<5xi32>, tensor<5xi32>, tensor<5xi32>) -> tensor<4x8x8x10x2xf32>
    
      // CHECK: mhlo.reshape %[[SLICE]]
      // CHECK-SAME: -> tensor<4x8x8x10x2xf32>
    
      func.return
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon May 06 18:46:23 UTC 2024
    - 335.5K bytes
    - Viewed (0)
Back to top