Searched refs:self_data_ptr (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/nested/cuda/ |
| D | NestedTensorBinaryOps.cu | 88 const scalar_t* self_data_ptr = self_buffer.const_data_ptr<scalar_t>(); in _nested_op_dense_esuhm_kernel() local
94 self_data_ptr, in _nested_op_dense_esuhm_kernel()
|
| /external/pytorch/aten/src/ATen/native/ |
| D | TensorAdvancedIndexing.cpp | 1423 auto self_data_ptr = self.const_data_ptr<scalar_t>(); in index_select_out_cpu_() local
1427 …[&index_contig, &numel, &self_numel, &self_data_ptr, &self_stride, &result_data_ptr, &result_strid… in index_select_out_cpu_()
1432 const scalar_t *self_ip = self_data_ptr + self_i * self_stride; in index_select_out_cpu_()
1443 auto self_data_ptr = self.const_data_ptr<scalar_t>(); in index_select_out_cpu_() local
1447 …[&index_contig, &numel, &self_numel, &self_data_ptr, &self_stride, &result_data_ptr, &result_strid… in index_select_out_cpu_()
1452 const scalar_t *self_ip = self_data_ptr + self_i * self_stride; in index_select_out_cpu_()
|