Home
last modified time | relevance | path

Searched refs:self_start (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/aten/src/ATen/native/cpu/
DDistanceOpsKernel.cpp149 const scalar_t * const self_start = self.const_data_ptr<scalar_t>(); in run_parallel_pdist() local
150 const scalar_t * const self_end = self_start + self.numel(); in run_parallel_pdist()
161 …parallel_for(0, combs, internal::GRAIN_SIZE / (16 * m), [p, self_start, self_end, n, m, res_start]… in run_parallel_pdist()
169 const scalar_t * self_i = self_start + i * m; in run_parallel_pdist()
170 const scalar_t * self_j = self_start + j * m; in run_parallel_pdist()
301 const scalar_t * const self_start = self.const_data_ptr<scalar_t>(); in run_backward_parallel_pdist() local
307 …ernal::GRAIN_SIZE / (8 * n * n), [p, n, m, gs, grad_start, dist_start, self_start, res_start](int6… in run_backward_parallel_pdist()
310 const scalar_t * self_l = self_start + l * Vec::size(); in run_backward_parallel_pdist()
319 …backward_down_column_pdist<F>(self_start + (m - remainder), res_start + (m - remainder), grad_star… in run_backward_parallel_pdist()
/external/python/cpython3/Objects/
Dbytearrayobject.c1315 const char *self_start = PyByteArray_AS_STRING(self); in bytearray_removeprefix_impl() local
1321 && memcmp(self_start, prefix_start, prefix_len) == 0) in bytearray_removeprefix_impl()
1323 return PyByteArray_FromStringAndSize(self_start + prefix_len, in bytearray_removeprefix_impl()
1327 return PyByteArray_FromStringAndSize(self_start, self_len); in bytearray_removeprefix_impl()
1347 const char *self_start = PyByteArray_AS_STRING(self); in bytearray_removesuffix_impl() local
1353 && memcmp(self_start + self_len - suffix_len, in bytearray_removesuffix_impl()
1356 return PyByteArray_FromStringAndSize(self_start, in bytearray_removesuffix_impl()
1360 return PyByteArray_FromStringAndSize(self_start, self_len); in bytearray_removesuffix_impl()
Dbytesobject.c2289 const char *self_start = PyBytes_AS_STRING(self); in bytes_removeprefix_impl() local
2296 && memcmp(self_start, prefix_start, prefix_len) == 0) in bytes_removeprefix_impl()
2298 return PyBytes_FromStringAndSize(self_start + prefix_len, in bytes_removeprefix_impl()
2306 return PyBytes_FromStringAndSize(self_start, self_len); in bytes_removeprefix_impl()
2326 const char *self_start = PyBytes_AS_STRING(self); in bytes_removesuffix_impl() local
2333 && memcmp(self_start + self_len - suffix_len, in bytes_removesuffix_impl()
2336 return PyBytes_FromStringAndSize(self_start, in bytes_removesuffix_impl()
2344 return PyBytes_FromStringAndSize(self_start, self_len); in bytes_removesuffix_impl()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc785 int64_t self_start = std::max<int64_t>(0, left_bound); in ExchangeHalo() local
787 if (self_start == 0 && self_limit == input_shard_size) { in ExchangeHalo()
791 self_shape.set_dimensions(dim, self_limit - self_start); in ExchangeHalo()
793 start_indices[dim] = self_start; in ExchangeHalo()