Searched defs:sizes_ptr (Results 1 – 10 of 10) sorted by relevance
84 const int64_t* sizes_ptr = sizes.const_data_ptr<int64_t>(); in construct_opt_sizes() local117 const int64_t* sizes_ptr = sizes.const_data_ptr<int64_t>(); in construct_nested_strides() local155 const int64_t* sizes_ptr = sizes.const_data_ptr<int64_t>(); in construct_offsets() local347 const int64_t* sizes_ptr = tensor.const_data_ptr<int64_t>(); in get_numel_from_nested_size_tensor() local
168 const int64_t* sizes_ptr) { in is_contiguous_strides_for_shape()281 const int64_t* sizes_ptr, in reinterpret_tensor_wrapper()294 const int64_t* sizes_ptr, in reinterpret_tensor_wrapper()
355 const int64_t* sizes_ptr, in aoti_torch__reinterpret_tensor()371 const int64_t* sizes_ptr, in aoti_torch_empty_strided()396 const int64_t* sizes_ptr, in aoti_torch_create_tensor_from_blob()423 const int64_t* sizes_ptr, in aoti_torch_create_tensor_from_blob_v2()1165 const int64_t* sizes_ptr, in aoti_torch__alloc_from_pool()
196 int64_t* const sizes_ptr = sizes.data_ptr<int64_t>(); in NestedTensor_batch_offsets_from_size_tensor() local237 auto* sizes_ptr = sizes.data_ptr<int64_t>(); in NestedTensor_to_mask() local
39 const auto sizes_ptr = sizes.data_ptr<int64_t>(); in NestedTensor_get_max_size_from_size_tensor() local
298 auto sizes_ptr = sizes_data_start; in NestedTensor_to_padded_tensor_generic() local
21 auto* sizes_ptr = sizes.data_ptr<int64_t>(); in get_nnz() local52 auto* sizes_ptr = sizes.data_ptr<int64_t>(); in cumulative_and_max_seq_len_nnz() local
166 auto* sizes_ptr = sizes.data_ptr<int64_t>(); in check_for_seq_len_0_and_consistent_head_dim_nested_tensor_helper() local
742 auto* sizes_ptr = sizes.data_ptr<int64_t>(); in check_for_seq_len_1_nested_tensor() local
455 const auto sizes_ptr = offsets_ptr + sizes.numel() + 1; in transform_bias_rescale_qkv_cuda() local