- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for NextIterationSourceOp (0.26 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
op->getResults()); } if (auto iter_sink = dyn_cast<tf_executor::NextIterationSinkOp>(op)) { auto iter_source = cast<tf_executor::NextIterationSourceOp>( iter_sink.getToken().getDefiningOp()); return RefineTypeForPassThroughOperands( op, iter_sink.getOperands().drop_front().take_front(), iter_source.getResults()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
mlir::OpBuilder builder_at_begin(builder_.getBlock(), builder_.getBlock()->begin()); auto source_op = builder_at_begin.create<mlir::tf_executor::NextIterationSourceOp>( loc, operands[0].getType(), result.attributes); return builder_.create<mlir::tf_executor::NextIterationSinkOp>( loc, source_op.getToken(), operands, result.attributes); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)