Searched defs:self_data_bytes (Results 1 – 4 of 4) sorted by relevance
63 const auto* self_data_bytes = data[1]; in cpu_cum_base_kernel() local236 …binary_kernel_reduce_lastdim(iter, [](char* result_data_bytes, char* self_data_bytes, int64_t size… in norm_kernel_tensor_iterator_impl()393 …binary_kernel_reduce_lastdim(iter, [&](char* result_data_bytes, char* self_data_bytes, int64_t siz… in argmax_kernel_impl()417 …binary_kernel_reduce_lastdim(iter, [&](char* result_data_bytes, char* self_data_bytes, int64_t siz… in argmin_kernel_impl()
212 auto* self_data_bytes = data[SELF_ITER_STRIDE_IDX]; in operator ()() local302 auto* self_data_bytes = data[SELF_ITER_STRIDE_IDX]; in operator ()() local398 auto* self_data_bytes = data[SELF_ITER_STRIDE_IDX]; in operator ()() local493 auto* self_data_bytes = data[SELF_ITER_STRIDE_IDX]; in operator ()() local589 auto* self_data_bytes = data[SELF_ITER_STRIDE_IDX]; in operator ()() local
204 auto* self_data_bytes = data[0]; in index_fill_kernel() local223 auto* self_data_bytes = data[0]; in index_fill_kernel() local262 auto* self_data_bytes = data[0]; in index_copy_kernel() local281 auto* self_data_bytes = data[0]; in index_copy_kernel() local
85 const auto* self_data_bytes = data[2]; in compare_base_kernel() local252 const auto* self_data_bytes = data[2]; in mode_kernel_impl() local