Function BroadcastInDim

tensorflow/compiler/xla/client/xla_builder.cc:2952–2957

}

XlaOp BroadcastInDim(const XlaOp operand,
                     const absl::Span<const int64> out_dim_size,
                     const absl::Span<const int64> broadcast_dimensions) {
  return operand.builder()->BroadcastInDim(operand, out_dim_size,
                                           broadcast_dimensions);
}

XlaOp Copy(const XlaOp operand) {
  return operand.builder()->UnaryOp(HloOpcode::kCopy, operand);