| /aosp_15_r20/external/pytorch/torch/_higher_order_ops/ |
| H A D | triton_kernel_wrap.py | 528 def __call__(self, kernel_idx, constant_args_idx, grid, kwargs): argument
545 def __call__(self, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone): argument
560 *, kernel_idx, constant_args_idx, grid, kwargs argument
582 mode, *, kernel_idx, constant_args_idx, grid, kwargs argument
589 def _(*, kernel_idx, constant_args_idx, grid, kwargs): argument
611 mode, *, kernel_idx, constant_args_idx, grid, kwargs argument
629 ctx, kernel_idx, constant_args_idx, grid, kwargs argument
667 *, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone argument
688 mode, *, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone argument
704 mode, *, kernel_idx, constant_args_idx, grid, kwargs, tensors_to_clone argument
[all …]
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| H A D | work_group_picking.cc | 36 int multiplier, int3 grid, const KernelInfo& kernel_info, in GenerateWorkGroupSizesXYMultipleOf()
66 int multiplier, int3 grid, const KernelInfo& kernel_info, in GenerateWorkGroupSizesXMultipleOf()
93 const KernelInfo& kernel_info, const int3& grid, in GetWorkGroupsAlignedToGrid()
198 int3 GetWorkGroupXY128ConvLinear(const int3& grid) { in GetWorkGroupXY128ConvLinear()
207 int3 GetWorkGroupXY128Conv(const int3& grid) { in GetWorkGroupXY128Conv()
216 int3 GetWorkGroupXY128Simple(const int3& grid) { return int3(16, 8, 1); } in GetWorkGroupXY128Simple()
218 int3 GetWorkGroup(const int3& grid, int max_size) { in GetWorkGroup()
226 int3 GetWorkGroupConv(const int3& grid, int max_size, int max_z_size) { in GetWorkGroupConv()
239 const int3& grid, in GetPossibleWorkGroupsXYMultipleOf()
248 const int3& grid, in GetPossibleWorkGroupsXMultipleOf()
[all …]
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_triton_kernels.py | 84 grid = lambda meta: (triton.cdiv(n_elements, meta["BLOCK_SIZE"]),) function
244 grid = lambda meta: (triton.cdiv(n_elements, meta["BLOCK_SIZE"]),) function
251 grid = lambda meta: (triton.cdiv(n_elements, meta["BLOCK_SIZE"]),) function
329 grid = lambda meta: (triton.cdiv(n_elements, meta["BLOCK_SIZE"]),) function
358 grid = lambda meta: (triton.cdiv(n_elements, meta["BLOCK_SIZE"]),) function
420 grid = lambda meta: (triton.cdiv(n_elements, meta["BLOCK_SIZE"]),) function
533 grid = lambda meta: (triton.cdiv(n_elements, meta["BLOCK_SIZE"]),) function
554 grid = lambda meta: (triton.cdiv(n_elements, meta["BLOCK_SIZE"]),) function
709 grid = lambda meta: (triton.cdiv(n_elements, meta["BLOCK_SIZE"]),) function
770 grid = lambda meta: ( function
[all …]
|
| /aosp_15_r20/external/replicaisland/src/com/replica/replicaisland/ |
| H A D | BufferLibrary.java | 36 public void add(Grid grid) { in add()
48 Grid grid = mGridList.get(x); in generateHardwareBuffers() local
58 Grid grid = mGridList.get(x); in releaseHardwareBuffers() local
68 Grid grid = mGridList.get(x); in invalidateHardwareBuffers() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | GridSamplerUtils.h | 24 const TensorBase& grid in check_grid_sampler_common()
63 const TensorBase& grid in check_grid_sampler_2d()
75 const TensorBase& grid, in check_grid_sampler_3d()
94 const TensorBase& grid in cond_cudnn_grid_sampler()
|
| H A D | GridSampler.cpp | 41 Tensor grid_sampler_3d_cpu_impl(const Tensor& input, const Tensor& grid, in grid_sampler_3d_cpu_impl()
206 const Tensor& input, const Tensor& grid, in grid_sampler_3d_backward_cpu_impl()
448 const Tensor& grid, in _grid_sampler_2d_cpu_quantized()
557 Tensor _grid_sampler_2d_cpu_fallback(const Tensor& input, const Tensor& grid, in _grid_sampler_2d_cpu_fallback()
713 const Tensor& input, const Tensor& grid, in _grid_sampler_2d_cpu_fallback_backward()
919 Tensor grid_sampler_2d_cpu(const Tensor& input, const Tensor& grid, in grid_sampler_2d_cpu()
963 Tensor grid_sampler_3d_cpu(const Tensor& input, const Tensor& grid, in grid_sampler_3d_cpu()
979 grid_sampler_2d_backward_cpu(const Tensor& grad_output, const Tensor& input, const Tensor& grid, in grid_sampler_2d_backward_cpu()
1028 grid_sampler_3d_backward_cpu(const Tensor& grad_output, const Tensor& input, const Tensor& grid, in grid_sampler_3d_backward_cpu()
1048 const Tensor& grid, in grid_sampler()
|
| H A D | AffineGridGenerator.cpp | 18 static at::Tensor linspace_from_neg_one(const Tensor& grid, int64_t num_steps, in linspace_from_neg_one()
72 auto grid = base_grid.view({N, H * W, 3}).bmm(theta.transpose(1, 2)); in affine_grid_generator_4D() local
85 auto grid = base_grid.view({N, D * H * W, 4}).bmm(theta.transpose(1, 2)); in affine_grid_generator_5D() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | GridSampler.cpp | 19 Tensor grid_sampler_2d_cuda(const Tensor& input, const Tensor& grid, in grid_sampler_2d_cuda()
31 Tensor grid_sampler_3d_cuda(const Tensor& input, const Tensor& grid, in grid_sampler_3d_cuda()
46 const Tensor& grid, int64_t interpolation_mode, int64_t padding_mode, in grid_sampler_2d_backward_cuda()
65 const Tensor& grid, int64_t interpolation_mode, int64_t padding_mode, in grid_sampler_3d_backward_cuda()
|
| H A D | GridSampler.cu | 29 TensorInfo<const scalar_t, index_t> grid, in grid_sampler_2d_kernel()
159 TensorInfo<const scalar_t, index_t> grid, in grid_sampler_3d_kernel()
316 TensorInfo<const scalar_t, index_t> grid, in grid_sampler_2d_backward_kernel()
522 TensorInfo<const scalar_t, index_t> grid, in grid_sampler_3d_backward_kernel()
751 const TensorBase &output, const TensorBase &input, const TensorBase &grid, in launch_grid_sampler_2d_forward_kernel()
795 const TensorBase &output, const TensorBase &input, const TensorBase &grid, in launch_grid_sampler_3d_forward_kernel()
842 const TensorBase &grid, int64_t interpolation_mode, int64_t padding_mode, in launch_grid_sampler_2d_backward_kernel()
905 const TensorBase& grid, int64_t interpolation_mode, int64_t padding_mode, in launch_grid_sampler_3d_backward_kernel()
|
| H A D | Sort.cu | 75 dim3 grid; in sort() local
139 dim3 grid; in sort() local
260 dim3 grid; in fixed_size_sort() local
|
| H A D | ScanUtils.cuh | 209 …dim3 grid(std::min(maxGridDim, num_orows), std::min(maxGridDim, ceil_div(num_irows, int64_t{thread… in scan_outer_dim_with_indices() local
231 …dim3 grid(std::min(at::cuda::getCurrentDeviceProperties()->maxGridSize[0], ceil_div(num_rows, int(… in scan_innermost_dim_with_indices() local
404 …dim3 grid(std::min(maxGridDim, num_orows), std::min(maxGridDim, ceil_div(num_irows, int64_t{thread… in scan_outer_dim() local
431 dim3 grid(std::min(maxGridDim, ceil_div(num_rows, int64_t{threads.y}))); in scan_innermost_dim() local
|
| H A D | SortStable.cu | 115 dim3 grid = GET_BLOCKS(nsort); in segmented_sort_large_segments() local
150 dim3 grid = GET_BLOCKS(numel); in segmented_sort_pairs_by_full_sort() local
198 dim3 grid = GET_BLOCKS(numel); in segmented_sort_pairs() local
|
| H A D | CUDALoops.cuh | 114 int64_t grid = (N + block_work_size() - 1) / block_work_size(); in launch_vectorized_kernel() local
161 int64_t grid = (N + block_work_size() - 1) / block_work_size(); in launch_unrolled_kernel() local
190 dim3 grid((N + block.x * vt - 1) / (block.x * vt)); in launch_legacy_kernel() local
|
| H A D | MultinomialKernel.cu | 92 dim3 grid(rows < numSM * 4 ? rows : numSM * 4); in renormRows() local
367 dim3 grid(std::min(static_cast<int>(numDist), numSM * 4)); in multinomial_with_replacement_kernel_impl() local
429 dim3 grid((n_sample-1)/block.x+1, grid_y); in multinomial_with_replacement_kernel_impl() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/runtime/ |
| H A D | triton_heuristics.py | 495 grid, argument
535 grid, argument
569 grid, argument
641 def bench(self, launcher, *args, grid, with_profiler=False, **kwargs): argument
738 def save_gpu_kernel(self, grid, stream, launcher): argument
829 def run(self, *args, grid, stream, **kwargs): argument
960 def run(self, *args, grid, stream): argument
1683 def grid(*numels): function
|
| /aosp_15_r20/packages/apps/Launcher3/tests/multivalentTests/src/com/android/launcher3/util/ |
| D | GridOccupancyTest.java | 22 GridOccupancy grid = initGrid(4, in testFindVacantCell() local
43 GridOccupancy grid = initGrid(4, in testIsRegionVacant() local
61 GridOccupancy grid = new GridOccupancy(cols, rows); in initGrid() local
|
| /aosp_15_r20/external/cronet/third_party/rust/chromium_crates_io/vendor/qr_code-2.0.0/src/decode/ |
| H A D | mod.rs | 25 pub grid: G, field
242 let grid = [ in test_rqrr() localVariable
328 let grid = [ in test_github() localVariable
467 let grid = [ in test_number() localVariable
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/functorch/ |
| H A D | BatchRulesModules.cpp | 113 grid_sample_batch_rule(const Tensor& input, std::optional<int64_t> input_bdim, const Tensor& grid, … in grid_sample_batch_rule()
142 const Tensor& grid, std::optional<int64_t> grid_bdim) { in grid_sample_backward_helper_in()
182 const Tensor& grid, std::optional<int64_t> grid_bdim, in grid_sample_backward_batch_rule()
202 const Tensor& grid, std::optional<int64_t> grid_bdim, in cudnn_grid_sample_backward_batch_rule()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| H A D | workgroup_selection.cc | 32 void AddCornerCases(const T& grid, int max_work_group_total_size, in AddCornerCases()
148 const T& grid, int min_work_group_total_size, int max_work_group_total_size, in GenerateWorkGroupSizes()
191 void GenerateWorkGroupSizesAlignedToGrid(const T& grid, in GenerateWorkGroupSizesAlignedToGrid()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cudnn/ |
| H A D | GridSampler.cpp | 65 void checkGridSize(CheckedFrom c, TensorArg grid, TensorArg input) { in checkGridSize()
90 TensorArg input{input_contig, "input", 1}, grid{grid_contig, "grid", 2}; in cudnn_grid_sampler_forward() local
141 TensorArg input{input_contig, "input", 1}, grid{grid_contig, "grid", 2}, in cudnn_grid_sampler_backward() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_wrapper_cuda.py | 72 grid, argument
119 grid, argument
349 grid=None, argument
|
| /aosp_15_r20/external/pigweed/pw_web/webconsole/styles/ |
| H A D | Home.module.scss | 53 .grid { class
75 .grid { class
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | Sleep.cu | 30 dim3 grid(1); in sleep() local
62 dim3 grid(at::cuda::getCurrentDeviceProperties()->multiProcessorCount * 60); in flush_icache() local
|
| /aosp_15_r20/frameworks/base/packages/EasterEgg/src/com/android/egg/widget/ |
| H A D | PaintChipsActivity.kt | 62 val grid = buildFullWidget(this, ClickBehavior.SHARE) in rebuildGrid() constant
|
| /aosp_15_r20/external/deqp/modules/gles3/functional/ |
| H A D | es3fVertexTextureTests.cpp | 565 …olors(const vector<Vec4> &quadColors, const Rect ®ion, const PosTexCoordQuadGrid<TexType> &grid, in setPixelColors()
610 … const PosTexCoordQuadGrid<TexType> &grid, tcu::Surface &dst, const Rect &dstRegion) in computeReference()
867 … const Grid grid(GRID_SIZE_2D, rend.region.size(), getTextureSize(*m_textures[rend.textureNdx]), in iterate() local
947 void Vertex2DTextureCase::computeReferenceCell(int textureNdx, float lod, const Grid &grid, tcu::Su… in computeReferenceCell()
1217 const Grid grid( in iterate() local
1299 void VertexCubeTextureCase::computeReferenceCell(int textureNdx, float lod, const Grid &grid, tcu::… in computeReferenceCell()
1550 …const Grid grid(GRID_SIZE_2D_ARRAY, rend.region.size(), getTextureSize(*m_textures[rend.textureNdx… in iterate() local
1630 void Vertex2DArrayTextureCase::computeReferenceCell(int textureNdx, float lod, const Grid &grid, tc… in computeReferenceCell()
1884 … const Grid grid(GRID_SIZE_3D, rend.region.size(), getTextureSize(*m_textures[rend.textureNdx]), in iterate() local
1965 void Vertex3DTextureCase::computeReferenceCell(int textureNdx, float lod, const Grid &grid, tcu::Su… in computeReferenceCell()
|