Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TensorListPopBack (0.26 sec)

  1. tensorflow/compiler/mlir/lite/tests/legalize-tensorlist.mlir

      %0, %1 = "tf.TensorListPopBack"(%arg0, %arg1) : (tensor<!tf_type.variant<tensor<2xi32>>>, tensor<1xi32>) -> (tensor<!tf_type.variant<tensor<2xi32>>>, tensor<2xi32>)
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 09:41:17 UTC 2024
    - 9.5K bytes
    - Viewed (0)
  2. 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(),
            "TensorListPopBack", TFL::ConstBytesAttr::get(getContext(), ""));
        rewriter.replaceOp(op, converted.getResults());
        return success();
      }
    };
    
    struct ConvertTensorListPushBack
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 10.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

        llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size) {
      auto buffer = pop.getInputHandle();
      auto it = buffer_to_size->find(buffer);
      if (it == buffer_to_size->end()) {
        pop.emitOpError("found tf.TensorListPopBack on unknown TensorList.");
        return failure();
      }
      if (it->getSecond().fixed) {
        return pop.emitError("cannot pop on a fixed-size tensor list");
      }
      auto size = it->getSecond().size;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      }];
    
      let arguments = (ins
        TF_VariantTensor:$input_handle
      );
    
      let results = (outs
        TF_Int32Tensor:$length
      );
    }
    
    def TF_TensorListPopBackOp : TF_Op<"TensorListPopBack", [Pure]> {
      let summary = [{
    Returns the last element of the input list as well as a list with all but that element.
      }];
    
      let description = [{
    Fails if the list is empty.
    
    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