Searched refs:calc_gpu_sizes (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/aten/src/ATen/native/vulkan/api/ |
| D | Tensor.cpp | 92 std::vector<int64_t> calc_gpu_sizes( in calc_gpu_sizes() function
231 gpu_sizes_{calc_gpu_sizes(sizes, memory_layout_, storage_type)},
262 gpu_sizes_{calc_gpu_sizes(sizes, memory_layout_, storage_type)},
402 gpu_sizes_ = calc_gpu_sizes(sizes_, memory_layout_, storage_type()); in update_size_metadata()
426 calc_gpu_sizes(new_sizes, memory_layout_, storage_type()), in reallocate()
|