Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 12 for TensorListPushBack (0.19 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/tensor_list_ops_decomposition.mlir

      %elem = "tf._SomeOp"() : () -> tensor<f32>
      // CHECK-NOT: "tf.TensorListPushBack"
      // CHECK: %[[UPDATE:.*]] = "tf.XlaDynamicUpdateSlice"
      // CHECK: %[[CONST1:.*]] = "tf.Const"() <{value = dense<1> : tensor<1xi32>}> : () -> tensor<1xi32>
      // CHECK: %[[ADD:.*]] = "tf.AddV2"(%[[BARG2]], %[[CONST1]])
      // CHECK-NOT: "tf.TensorListPushBack"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 38.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/tests/legalize-tensorlist.mlir

      %0 = "tf.TensorListPushBack"(%arg0, %arg1) : (tensor<!tf_type.variant<tensor<?x1xf32>>>, tensor<16x1xf32>) -> tensor<!tf_type.variant<tensor<?x1xf32>>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir

        %tl_0 = "tf.EmptyTensorList"(%elem_shape, %size) : (tensor<2xi32>, tensor<i32>) -> tensor<!tf_type.variant<tensor<?x1xf32>>>
        %tl_1 = "tf.TensorListPushBack"(%tl_0, %elem) : (tensor<!tf_type.variant<tensor<?x1xf32>>>, tensor<16x1xf32>) -> tensor<!tf_type.variant<tensor<?x1xf32>>>
        %tl_2 = "tf.TensorListPushBack"(%tl_1, %elem) : (tensor<!tf_type.variant<tensor<?x1xf32>>>, tensor<16x1xf32>) -> tensor<!tf_type.variant<tensor<?x1xf32>>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jan 23 17:24:10 UTC 2024
    - 167.4K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc

        // op into an op with a single varidic output in `.td`.
        auto converted = rewriter.create<TFL::CustomOp>(
            op->getLoc(), op->getResultTypes(), op->getOperands(),
            "TensorListPushBack", TFL::ConstBytesAttr::get(getContext(), ""));
        rewriter.replaceOp(op, converted.getResults());
        return success();
      }
    };
    
    struct ConvertVariantAddNOp : public OpRewritePattern<TF::AddNOp> {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 10.6K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/tests/lower-static-tensor-list.mlir

    // CHECK:      return [[RESULT]] : tensor<?x?x?xf32>
    }
    
    // -----
    
    func.func @tensorlistPushBack(%arg0: tensor<3x10xf32>, %arg1: tensor<1xi32>, %arg2: tensor<10xf32>) -> tensor<?x10xf32> {
      %0 = "tf.TensorListFromTensor"(%arg0, %arg1) : (tensor<3x10xf32>, tensor<1xi32>) -> tensor<!tf_type.variant<tensor<10xf32>>>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 39.9K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

      auto buffer = push.getInputHandle();
      auto it = buffer_to_size->find(buffer);
      if (it == buffer_to_size->end()) {
        return push.emitOpError(
            "found tf.TensorListPushBack on unknown TensorList.");
      }
      if (it->getSecond().fixed) {
        return push.emitError("cannot push on a fixed-size tensor list");
      }
      auto size = it->getSecond().size;
      OpBuilder builder(push);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  7. tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc

            // TODO(srbs): This filter is a little conservative. E.g. a subgraph of
            // the form:
            //                          Const
            //                            |
            // EmptyTensorList -> TensorListPushBack -> TensorListPopBack -> Op
            //                                                  |
            //                                        (Discard popped list)
            //
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 22 08:47:20 UTC 2024
    - 51K bytes
    - Viewed (0)
  8. tensorflow/compiler/mlir/tensorflow/transforms/tf_passes.td

      %max_size = "tf.Const"() {value = dense<10> : tensor<i32>} : () -> tensor<i32>
      %tl = "tf.EmptyTensorList"(%elem_shape, %max_size) : (tensor<2xi32>, tensor<i32>) -> tensor<!tf_type.variant<tensor<8x4xf32>>>
      %push = "tf.TensorListPushBack"(%tl, %arg0) : (tensor<!tf_type.variant<tensor<8x4xf32>>>, tensor<8x4xf32>) -> tensor<!tf_type.variant<tensor<8x4xf32>>>
      return
    }
    ```
    
    will be transformed to:
    
    ```mlir
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:18:05 UTC 2024
    - 99.6K bytes
    - Viewed (0)
  9. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

          for (const mlir::OpOperand &use : arg.getUses()) {
            mlir::Operation *op = use.getOwner();
            // Currently we only check if the tensorlist argument is consumed by
            // `TensorListPushBack` or `TensorListResize`, since those are the only
            // length-mutating ops supported in this pass.
            if (llvm::isa<TF::TensorListPushBackOp>(op) ||
                llvm::isa<TF::TensorListResizeOp>(op)) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  10. tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md

      %max_size = "tf.Const"() {value = dense<10> : tensor<i32>} : () -> tensor<i32>
      %tl = "tf.EmptyTensorList"(%elem_shape, %max_size) : (tensor<2xi32>, tensor<i32>) -> tensor<!tf_type.variant<tensor<8x4xf32>>>
      %push = "tf.TensorListPushBack"(%tl, %arg0) : (tensor<!tf_type.variant<tensor<8x4xf32>>>, tensor<8x4xf32>) -> tensor<!tf_type.variant<tensor<8x4xf32>>>
      return
    }
    ```
    
    will be transformed to:
    
    ```mlir
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Aug 02 02:26:39 UTC 2023
    - 96.4K bytes
    - Viewed (0)
Back to top