Searched refs:self_buffer (Results 1 – 7 of 7) sorted by relevance
| /external/pytorch/aten/src/ATen/native/nested/ |
| D | NestedTensorBackward.cpp | 126 const Tensor& self_buffer = nt_self->get_buffer(); in _nested_sum_backward_cpu() local
130 const Tensor& self_grad_buffer = self_buffer.new_empty(self_buffer.sizes()); in _nested_sum_backward_cpu()
169 const Tensor& self_buffer = nt_self->get_buffer(); in _nested_select_backward_symint() local
171 const Tensor& self_grad_buffer = self_buffer.new_zeros(self_buffer.sizes()); in _nested_select_backward_symint()
|
| D | NestedTensorMatmul.cpp | 31 …const Tensor& self_buffer = self.is_nested() ? get_nested_tensor_impl(self)->get_unsafe_storage_as… in bmm_nested() local
56 …Tensor out_buffer = self.is_nested() ? self_buffer.new_empty(out_numel) : mat2_buffer.new_empty(ou… in bmm_nested()
66 …self_buffer.as_strided(get_size_for_index(self, i), get_stride_for_index(self, i), get_offset_for_… in bmm_nested()
|
| D | NestedTensorBinaryOps.cpp | 130 const auto self_buffer = self_ptr->get_buffer(); in NestedTensor_elementwise_Tensor() local
132 auto result_buffer = at::empty_like(self_buffer); in NestedTensor_elementwise_Tensor()
|
| D | NestedTensorFactories.cpp | 151 const Tensor& self_buffer = self_ptr->get_unsafe_storage_as_tensor(), in clone_nested() local
153 Tensor output_buffer = at::empty(self.numel(), self_buffer.options()); in clone_nested()
|
| D | NestedTensorTransformerFunctions.cpp | 78 const Tensor& self_buffer = nt_self->get_buffer(); in NestedTensor_matmul() local
80 at::mm(self_buffer.reshape({-1, other.sizes()[0]}), other); in NestedTensor_matmul()
|
| /external/pytorch/aten/src/ATen/native/nested/cuda/ |
| D | NestedTensorBinaryOps.cu | 79 const auto self_buffer = self_ptr->get_buffer(); in _nested_op_dense_esuhm_kernel() local
88 const scalar_t* self_data_ptr = self_buffer.const_data_ptr<scalar_t>(); in _nested_op_dense_esuhm_kernel()
|
| D | NestedTensorMatmul.cu | 331 …const Tensor &self_buffer = self.is_nested() ? get_nested_tensor_impl(self)->get_unsafe_storage_as… in bmm_nested_cuda() local
334 Tensor out_buffer = self_buffer.new_empty(out_numel); in bmm_nested_cuda()
362 aptr[i] = self_buffer.data_ptr<scalar_t>() + get_offset_for_index(self, i); in bmm_nested_cuda()
400 …self_buffer.as_strided(get_size_for_index(self, i), get_stride_for_index(self, i), get_offset_for_… in bmm_nested_cuda()
|