Searched defs:view_sizes (Results 1 – 4 of 4) sorted by relevance
21 std::vector<int64_t>& view_sizes) { in compute_out_sizes()51 IntListPtr view_sizes = graph->get_int_list(extra_args[0]); in resize_view_node() local
282 std::vector<int64_t> view_sizes = std::vector<int64_t>( in get_chunk_cat_out_sizes() local591 std::vector<int64_t> view_sizes = get_chunk_cat_out_sizes( in _chunk_cat_out_cuda_contiguous() local786 auto view_sizes = detail::get_chunk_cat_out_sizes( in _chunk_cat_cuda() local
1728 auto view_sizes = index.sizes().vec(); in _scatter_via_index_put() local1741 auto view_sizes = index.sizes().vec(); in _scatter_via_index_put() local
2740 std::vector<int64_t> view_sizes(tensor_size.begin(), tensor_size.begin()+dim); in _pad_chunk() local