Searched defs:broadcast_to_shape (Results 1 – 2 of 2) sorted by relevance
80 const exec_aten::ArrayRef<Tensor::SizesType> broadcast_to_shape) { in tensor_is_broadcastable_to()143 auto broadcast_to_shape = broadcast_to.sizes(); in broadcast_tensor() local
57 llvm::SmallVector<int64_t, 4> broadcast_to_shape; in ExpandOp() local