- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for TensorListPopBack (0.26 sec)
-
tensorflow/compiler/mlir/lite/tests/legalize-tensorlist.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 09:41:17 UTC 2024 - 9.5K bytes - Viewed (0) -
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) -
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) -
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)