Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for GetNumReplicatedBlockArguments (0.24 sec)

  1. 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)
Back to top