Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TensorListPushBack (0.36 sec)

  1. 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)
  2. 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)
  3. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

        TF_VariantTensor:$output_handle,
        TF_Tensor:$tensor
      );
    
      TF_DerivedResultTypeAttr element_dtype = TF_DerivedResultTypeAttr<1>;
    }
    
    def TF_TensorListPushBackOp : TF_Op<"TensorListPushBack", [Pure]> {
      let summary = [{
    Returns a list which has the passed-in `Tensor` as last element and the other elements of the given list in `input_handle`.
      }];
    
      let description = [{
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top