Searched defs:inp_ptr (Results 1 – 3 of 3) sorted by relevance
55 const uint8_t* inp_ptr = input + col_offset; in ColMeanAndVariance() local160 const uint8_t* inp_ptr = input + col_offset; in MinAndMax() local216 const uint8_t* inp_ptr = input + col_offset; in InstanceNorm() local
78 const scalar_t *inp_ptr = input.const_data_ptr<scalar_t>(); in grid_sampler_3d_cpu_impl() local271 const scalar_t *inp_ptr = input.const_data_ptr<scalar_t>(); in grid_sampler_3d_backward_cpu_impl() local490 uint8_t* inp_ptr = (uint8_t*)input.data_ptr<quint8>(); in _grid_sampler_2d_cpu_quantized() local592 const scalar_t *inp_ptr = input.const_data_ptr<scalar_t>(); in _grid_sampler_2d_cpu_fallback() local761 const scalar_t *inp_ptr = input.const_data_ptr<scalar_t>(); in _grid_sampler_2d_cpu_fallback_backward() local
200 def relu_kernel_(inp_ptr, out_ptr, sz, BLOCK_SIZE: tl.constexpr): argument