- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for start_step_0 (0.09 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
// that we'll execute at least two steps. builder.setInsertionPointAfter(non_tpu_caller_0); TF::StatefulPartitionedCallOp start_step_0; result = StartStep0(builder, loc, symbol_table, metadata_op, compilation_op, C_0, orig_callers, loop_operands_0, start_step_0); if (failed(result)) return signalPassFailure(); // Save the results of the forward_0 and core_tpu_0 calls by slicing them
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/embedding_pipelining.mlir
// CHECK: {{.*StatefulPartitionedCall.* f = @non_tpu.*}} // CHECK: {{.*StatefulPartitionedCall.* f = @start_step_0.*}} // CHECK: {{.*StatefulPartitionedCall.* f = @while_cond.*}} // CHECK: {{.*StatefulPartitionedCall.* f = @non_tpu.*}} // CHECK: {{.*StatefulPartitionedCall.* f = @start_step_1.*}} // CHECK: {{.*StatefulPartitionedCall.* f = @while_cond.*}}
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 33.1K bytes - Viewed (0)