- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for getInputHandle (0.43 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
auto it = buffer_to_size.find(length.getInputHandle()); if (it == buffer_to_size.end()) { length.emitOpError("found tf.TensorListLength on unknown TensorList."); return failure(); } OpBuilder builder(length); if (it->getSecond().fixed) { auto dim = cutil::CreateScalarConst( mlir::cast<RankedTensorType>(length.getInputHandle().getType()) .getDimSize(0),
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/lite/transforms/lower_static_tensor_list.cc
parent = identity.getInput(); } else if (auto set_item = parent.getDefiningOp<TF::TensorListSetItemOp>()) { parent = set_item.getInputHandle(); } else { break; } } if (auto block_arg = parent.dyn_cast<mlir::BlockArgument>()) { return block_arg.getArgNumber(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
// Checks that the input is created by TensorListFromTensorOp and the input // is only used by TensorListGetItemOp. auto tensor_list_from_tensor_op = dyn_cast_or_null<TensorListFromTensorOp>( op.getInputHandle().getDefiningOp()); if (!tensor_list_from_tensor_op || llvm::any_of( tensor_list_from_tensor_op->getUsers(), [](Operation *user) { return !isa<TensorListGetItemOp>(user); })) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)