Searched defs:out_ndims (Results 1 – 5 of 5) sorted by relevance
25 auto out_ndims = safe_downcast<uint32_t>(repeats.size()); in repeat() local
46 auto out_ndims = safe_downcast<uint32_t>(out_sizes.size()); in masked_fill_scalar() local
36 uvec4 out_ndims; in permute_4d() member
37 uvec4 out_ndims; in transpose_4d() member
79 ivec4 out_ndims; in add_permute_node() member