- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for NextIteration (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
// "NextIteration" node to a "Merge" node add cyclic dependencies and make the // topological sorting impossible. We need to remove these edges from the // input graph to infer shapes and construct a Function. For each // "NextIteration" node, there are two operations, "NextIteration.source" // and "NextIteration.sink" are added to the MLIR module.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
func.func @next_iteration_user(%arg0: tensor<32x?x256x4xf32>) -> tensor<?x?x?xf32> { %0 = tf_executor.graph { // CHECK: tf_executor.NextIteration.Source // CHECK-SAME: : tensor<32x?x4xf32> %1:3 = tf_executor.NextIteration.Source : tensor<?x?x?xf32> %out, %c_out = tf_executor.island { %dims = "tf.Const"() {value = dense<[32, -1, 4]> : tensor<3xi32>} : () -> tensor<3xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0)