Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 13 for ellipsis_mask (0.39 sec)

  1. tensorflow/compiler/mlir/lite/utils/lstm_utils.cc

                               ArrayRef<int32_t> end, ArrayRef<int32_t> strides,
                               int64_t begin_mask, int64_t end_mask,
                               int64_t ellipsis_mask, int64_t new_axis_mask,
                               int64_t shrink_axis_mask, OpBuilder* builder) {
      auto output_type = RankedTensorType::get(
          output_shape,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 36.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tests/quantize-variables.mlir

      %6 = "tfl.dequantize"(%5) : (tensor<1x34x1x3x!quant.uniform<i8:f32, 1.0>>) -> tensor<1x34x1x3xf32>
      %7 = "tfl.strided_slice"(%6, %cst_1, %cst_0, %cst) {begin_mask = 13 : i32, ellipsis_mask = 0 : i32, end_mask = 15 : i32, new_axis_mask = 0 : i32, shrink_axis_mask = 0 : i32, offset = false} : (tensor<1x34x1x3xf32>, tensor<4xi32>, tensor<4xi32>, tensor<4xi32>) -> tensor<1x2x1x3xf32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 20.3K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tfrt/tests/analysis/cost_analysis.mlir

        // expected-remark@+1 {{Cost: 1}}
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Aug 14 15:35:49 UTC 2023
    - 12.2K bytes
    - Viewed (0)
  4. tensorflow/cc/gradients/array_grad.cc

      int64_t begin_mask;
      int64_t end_mask;
      int64_t ellipsis_mask;
      int64_t new_axis_mask;
      int64_t shrink_axis_mask;
      TF_RETURN_IF_ERROR(
          GetNodeAttr(op.node()->attrs(), "begin_mask", &begin_mask));
      TF_RETURN_IF_ERROR(GetNodeAttr(op.node()->attrs(), "end_mask", &end_mask));
      TF_RETURN_IF_ERROR(
          GetNodeAttr(op.node()->attrs(), "ellipsis_mask", &ellipsis_mask));
      TF_RETURN_IF_ERROR(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 10 23:33:32 UTC 2023
    - 31.7K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tf2xla/api/v2/testdata/outside_compilation.mlir

          %outputs_55, %control_56 = tf_executor.island wraps "tf.StridedSlice"(%outputs, %outputs_49, %outputs_51, %outputs_53) {begin_mask = 0 : i64, device = "/device:CPU:0", ellipsis_mask = 0 : i64, end_mask = 0 : i64, new_axis_mask = 0 : i64, shrink_axis_mask = 1 : i64} : (tensor<4xi32>, tensor<1xi32>, tensor<1xi32>, tensor<1xi32>) -> tensor<*xi32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 19 20:19:45 UTC 2023
    - 21.9K bytes
    - Viewed (0)
  6. 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)
  7. 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)
  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/lite/tests/flatbuffer2mlir/test_schema.fbs

    }
    
    table SplitOptions {
      num_splits: int;
    }
    
    table SplitVOptions {
      num_splits: int;
    }
    
    table StridedSliceOptions {
      begin_mask: int;
      end_mask: int;
      ellipsis_mask: int;
      new_axis_mask: int;
      shrink_axis_mask: int;
    }
    
    table LogSoftmaxOptions {
    }
    
    table CastOptions {
      in_data_type: TensorType;
      out_data_type: TensorType;
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 19 19:46:06 UTC 2021
    - 26.1K bytes
    - Viewed (0)
Back to top