Home
last modified time | relevance | path

Searched refs:spatial_offset (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DGridSamplerKernel.cpp1048 for (int64_t spatial_offset = 0; spatial_offset < total_size; spatial_offset += step) { in grid_sample_2d_grid_slice_iterator() local
1049 auto grid_offset = spatial_offset * 2; in grid_sample_2d_grid_slice_iterator()
1050 auto len = std::min(step, total_size - spatial_offset); in grid_sample_2d_grid_slice_iterator()
1065 apply_fn(x, y, spatial_offset, len); in grid_sample_2d_grid_slice_iterator()
1108 int64_t spatial_offset = 0; in grid_sample_2d_grid_slice_iterator() local
1129 spatial_offset, len); in grid_sample_2d_grid_slice_iterator()
1133 spatial_offset += len; in grid_sample_2d_grid_slice_iterator()
1168 int64_t spatial_offset, int64_t len) { \ in grid_sampler_2d_cpu_kernel_impl()
1169 grid_sample.forward(out_slice, inp_slice, spatial_offset, \ in grid_sampler_2d_cpu_kernel_impl()
1254 int64_t spatial_offset, int64_t len) { \ in grid_sampler_2d_backward_cpu_kernel_impl()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/
H A Dbatch_to_space_nd.h75 const int spatial_offset = in_batch / output_batch_size; in BatchToSpaceND() local
78 spatial_offset / block_shape_width - crops_top; in BatchToSpaceND()
84 spatial_offset % block_shape_width - crops_left; in BatchToSpaceND()
/aosp_15_r20/external/libpalmrejection/ui/events/ozone/evdev/touch_filter/palm_model/
H A Donedevice_train_palm_detection_filter_inference_beta.cc471 const int32_t spatial_offset = in_batch / output_batch_size; in BatchToSpaceNDRank4() local
474 spatial_offset / block_shape_width - crops_top; in BatchToSpaceNDRank4()
480 spatial_offset % block_shape_width - crops_left; in BatchToSpaceNDRank4()
H A Donedevice_train_palm_detection_filter_inference.cc472 const int32_t spatial_offset = in_batch / output_batch_size; in BatchToSpaceNDRank4() local
475 spatial_offset / block_shape_width - crops_top; in BatchToSpaceNDRank4()
481 spatial_offset % block_shape_width - crops_left; in BatchToSpaceNDRank4()
H A Donedevice_train_palm_detection_filter_inference_v2.cc471 const int32_t spatial_offset = in_batch / output_batch_size; in BatchToSpaceNDRank4() local
474 spatial_offset / block_shape_width - crops_top; in BatchToSpaceNDRank4()
480 spatial_offset % block_shape_width - crops_left; in BatchToSpaceNDRank4()
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Doptimized_ops.h4241 const int spatial_offset = in_batch / output_batch_size; in BatchToSpaceND() local
4246 GetIndexRange(spatial_offset / block_shape_width - crops_top, in BatchToSpaceND()
4252 spatial_offset / block_shape_width - crops_top; in BatchToSpaceND()
4259 GetIndexRange(spatial_offset % block_shape_width - crops_left, in BatchToSpaceND()
4265 spatial_offset % block_shape_width - crops_left; in BatchToSpaceND()