Searched refs:rectangle_size (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/aten/src/ATen/native/cuda/ |
| D | TensorFactories.cu | 290 int64_t rectangle_size = 0; in tril_indices_cuda() local
292 rectangle_size = (row - rectangle_row_offset) * col; in tril_indices_cuda()
310 tril_size - rectangle_size, in tril_indices_cuda()
325 int64_t rectangle_size, in triu_indices_kernel() argument
331 if (linear_index < rectangle_size) { in triu_indices_kernel()
338 m_first_row, linear_index - rectangle_size, r, c); in triu_indices_kernel()
339 r += rectangle_size / col; in triu_indices_kernel()
367 int64_t rectangle_size = 0; in triu_indices_cuda() local
369 rectangle_size = std::min<int64_t>(row, -offset) * col; in triu_indices_cuda()
388 rectangle_size, in triu_indices_cuda()
|
| /external/pytorch/torch/_refs/ |
| D | __init__.py | 5786 rectangle_size = max(0, diff_row * col)
5788 return trapezoid_size, rectangle_size, m_first_row
5822 trapezoid_size, rectangle_size, m_first_row = _get_tril_sizes(row, col, offset)
5838 xs2 = arange_kw(0, rectangle_size, dtype=dtype)
5858 rectangle_size = max(0, min(row, -offset) * col)
5863 trapezoid_size = triu_size - rectangle_size
5865 return trapezoid_size, rectangle_size, m_first_row
5882 trapezoid_size, rectangle_size, m_first_row = _get_triu_sizes(row, col, offset)
5890 xs2 = arange_kw(0, rectangle_size, dtype=dtype)
5903 row_inds1 = row_inds1 + (rectangle_size // col)
|
| /external/deqp/external/openglcts/modules/gl/ |
| D | gl4cBufferStorageTests.cpp | 3437 static const size_t rectangle_size = 2 * 2 * sizeof(GLfloat); /* 2 * vec2 */ in iterate() local
3438 static const size_t rectangles_data_size = n_rectangles * rectangle_size; in iterate()
|