- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TensorListPopBack (0.81 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tensor_list_ops_decomposition.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 38.6K 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)