- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for TensorListPushBackOp (0.59 sec)
-
tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc
rewriter.replaceOp(op, converted.getResults()); return success(); } }; struct ConvertTensorListPushBack : public OpRewritePattern<TF::TensorListPushBackOp> { using OpRewritePattern::OpRewritePattern; LogicalResult matchAndRewrite(TF::TensorListPushBackOp op, PatternRewriter& rewriter) const override { // It is currently not possible to easily pack the output of a multi-result
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/lite/transforms/lower_static_tensor_list.cc
return CreateI32SplatConst(op.getLoc(), rewriter, {1}, 0); } }; struct ConvertTensorListPushBack : public OpConversionPattern<TF::TensorListPushBackOp> { using OpConversionPattern::OpConversionPattern; LogicalResult matchAndRewrite( TF::TensorListPushBackOp op, OpAdaptor adaptor, ConversionPatternRewriter &rewriter) const override { Value input_handle = adaptor.getOperands()[0];
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/transforms/tensor_list_ops_decomposition.cc
(*buffer_to_size)[buffer] = {size, /*fixed=*/true}; list.getOutputHandle().replaceAllUsesWith(buffer); list.erase(); return success(); } LogicalResult HandleTensorListPushBackOp( TF::TensorListPushBackOp push, llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size) { auto buffer = push.getInputHandle(); auto it = buffer_to_size->find(buffer); if (it == buffer_to_size->end()) { return push.emitOpError(
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/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc
GET_OPERATION_NAME(mlir::TF::EmptyTensorListOp), GET_OPERATION_NAME(mlir::TF::TensorListReserveOp), GET_OPERATION_NAME(mlir::TF::TensorListFromTensorOp), GET_OPERATION_NAME(mlir::TF::TensorListPushBackOp), GET_OPERATION_NAME(mlir::TF::TensorListPopBackOp), GET_OPERATION_NAME(mlir::TF::TensorListGetItemOp), GET_OPERATION_NAME(mlir::TF::TensorListSetItemOp),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
// (conceptually) modify it i.e., ops that take an input TensorList and // produce an output TensorList. for (auto& use : tensorlist.getUses()) { if (auto push = llvm::dyn_cast<TensorListPushBackOp>(use.getOwner())) { auto element_type = mlir::dyn_cast<RankedTensorType>(push.getTensor().getType()); if (!verify_and_update_potential_element_type(element_type)) return false;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0)