Searched defs:broadcast_size (Results 1 – 4 of 4) sorted by relevance
96 const auto broadcast_size = out.size(out.dim() - 1); in handle_last_dim_broadcast() local150 int64_t broadcast_size; in handle_broadcast_mul() local
44 int broadcast_size = size[0]; in Slice() local
285 std::vector<int64_t> broadcast_size(const Tensor& t1, const Tensor& t2) { in broadcast_size() function
870 int broadcast_size, in is_int_or_symint_list()