Searched refs:new_sizes_ptr (Results 1 – 2 of 2) sorted by relevance
200 auto new_sizes_ptr = new_sizes.data_ptr<int64_t>(); in matmul_nested_with_broadcasted_dense() local202 new_sizes_ptr[i * component_dim + 2] = E; in matmul_nested_with_broadcasted_dense()
1053 auto new_sizes_ptr = new_sizes.data_ptr<int64_t>(); in cat_nested_as_jagged() local1055 new_sizes_ptr[i * component_dim + (dim - 1)] = new_dim_size; in cat_nested_as_jagged()