Searched refs:result_stride_bytes (Results 1 – 2 of 2) sorted by relevance
1351 auto result_stride_bytes = result.stride(dim) * elementSize(result.scalar_type()); in index_select_out_cpu_() local1367 &result_stride_bytes](int64_t start, int64_t end) { in index_select_out_cpu_()1371 &resultSlice_data, &result_stride_bytes] () { in index_select_out_cpu_()1377 auto result_data = static_cast<char*>(resultSlice_data) + i * result_stride_bytes; in index_select_out_cpu_()1397 … &self_stride_bytes, &resultSlice_data, &result_stride_bytes](int64_t start, int64_t end) { in index_select_out_cpu_()1400 &self_stride_bytes, &resultSlice_data, &result_stride_bytes, &start, &end] () { in index_select_out_cpu_()1406 auto result_data = static_cast<char*>(resultSlice_data) + i * result_stride_bytes; in index_select_out_cpu_()
618 auto result_stride_bytes = result.stride(dim) * elementSize(result.scalar_type()); in TORCH_IMPL_FUNC() local633 auto result_data = static_cast<char*>(result_slice_data) + offset * result_stride_bytes; in TORCH_IMPL_FUNC()