- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for GetNumReplicatedBlockArguments (0.24 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/xla_broadcast.cc
<< ") to append to replicate to be num_replicas (" << replicate.getN() << ")"; } // add block arg to region. This is in $body. unsigned block_arg_idx = replicate.GetNumReplicatedBlockArguments(); Block& block = replicate.GetBody(); block_arg = block.insertArgument(block_arg_idx, type, replicate.getLoc()); // add to $replicated_inputs. This also updates OperandSegmentSizeAttr.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 13 18:52:07 UTC 2024 - 13.9K bytes - Viewed (0)