Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 41 for ellipsis_mask (0.47 sec)

  1. tensorflow/compiler/mlir/lite/tests/fuse-tftext.mlir

      %26 = "tf.AddV2"(%25, %13) {device = ""} : (tensor<1xi64>, tensor<i64>) -> tensor<1xi64>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 460.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tests/prepare-tf.mlir

      // CHECK-SAME: begin_mask = 7 : i64
      // CHECK-SAME: ellipsis_mask = 0 : i64
      // CHECK-SAME: end_mask = 14 : i64
      // CHECK-SAME: new_axis_mask = 0 : i64
      // CHECK-SAME: shrink_axis_mask = 0 : i64
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 29 07:26:59 UTC 2024
    - 59.8K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/metrics/testdata/strided_slice.mlir

      %57 = "tf.Const"() {value = dense<0> : tensor<4xi32>} : () -> tensor<4xi32>
      %534 = "tf.StridedSlice"(%arg0, %57, %57, %18) {begin_mask = 11 : i64, device = "", ellipsis_mask = 0 : i64, end_mask = 11 : i64, new_axis_mask = 4 : i64, shrink_axis_mask = 0 : i64} : (tensor<*xf32>, tensor<4xi32>, tensor<4xi32>, tensor<4xi32>) -> tensor<*xf32>
      "func.return"(%534) : (tensor<*xf32>) -> ()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Mar 22 14:25:57 UTC 2022
    - 536 bytes
    - Viewed (0)
  4. 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)
  5. 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)
  6. 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)
  7. 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)
  8. 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)
  9. tensorflow/compiler/mlir/lite/experimental/tac/tests/raise-target-subgraphs.mlir

        %0 = "tfl.shape"(%arg2) {tac.device = "DARWINN", tac.inference_type = "FLOAT"} : (tensor<?x?xi32>) -> tensor<2xi32>
        %1 = "tfl.strided_slice"(%0, %cst_3, %cst_2, %cst_2) {begin_mask = 0 : i32, ellipsis_mask = 0 : i32, end_mask = 0 : i32, new_axis_mask = 0 : i32, shrink_axis_mask = 1 : i32, offset = false, tac.device = "DARWINN", tac.inference_type = "FLOAT"} : (tensor<2xi32>, tensor<1xi32>, tensor<1xi32>, tensor<1xi32>) -> tensor<i32>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 74.9K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc

          return op.emitOpError("requires non-zero strides");
      }
    
      // Use bit compares to ensure ellipsis_mask is 0 or a power of 2, i.e. there
      // exists only no more than one ellipsis.
      uint32_t ellipsis_mask = op.getEllipsisMask();
      if (ellipsis_mask != 0 && !llvm::isPowerOf2_32(ellipsis_mask))
        return op.emitOpError("cannot have multiple ellipses");
    
      return success();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
Back to top