Searched defs:offsets_ptr (Results 1 – 13 of 13) sorted by relevance
90 int64_t *offsets_ptr = offsets.data_ptr<int64_t>(); in chunk_nested_tensor() local145 int64_t *offsets_ptr = offsets.data_ptr<int64_t>(); in split_with_sizes_nested() local
188 int64_t *offsets_ptr = self_ptr->get_storage_offsets().data_ptr<int64_t>(); in NestedTensor_unbind() local
204 int64_t* offsets_ptr = get_nested_tensor_impl(tensor) in get_offset_for_index() local
198 int32_t* const offsets_ptr = offsets.mutable_data_ptr<int32_t>(); in NestedTensor_batch_offsets_from_size_tensor() local
431 int64_t *offsets_ptr = self_ptr->get_storage_offsets().data_ptr<int64_t>(); in select_nested() local
70 auto offsets_ptr = input_size_ptr + padded_sizes_tensor.numel(); in nested_from_padded_cuda() local126 int64_t* offsets_ptr = offsets.mutable_data_ptr<int64_t>(); in batch_offsets_from_efficient_size() local
231 const int64_t* offsets_ptr = in nested_tensor_impl_is_contiguous() local
149 int64_t *offsets_ptr = offsets.mutable_data_ptr<int64_t>(); in construct_offsets() local
221 sljit_u8 *offsets_ptr = offsets; in call_with_args() local
202 offsets_ptr, argument
309 const auto offsets_ptr = reinterpret_cast<const int32_t *>(offsets_i.ptr()); in scale_nearest_nchw() local
304 int64_t* offsets_ptr = offsets.data_ptr<int64_t>(); in compute_pool_max() local
454 const auto offsets_ptr = metadata.data_ptr<int>(); in transform_bias_rescale_qkv_cuda() local