Searched refs:get_unsafe_storage_as_tensor (Results 1 – 8 of 8) sorted by relevance
| /external/pytorch/aten/src/ATen/native/nested/ |
| D | NestedTensorFactories.cpp | 62 at::empty_like(self_nt->get_unsafe_storage_as_tensor(), options); in empty_like_nested()
139 const Tensor& buffer = self_ptr->get_unsafe_storage_as_tensor(), in clone_nested()
151 const Tensor& self_buffer = self_ptr->get_unsafe_storage_as_tensor(), in clone_nested()
211 auto buffer = nt_impl->get_unsafe_storage_as_tensor(); in narrow_nested_symint()
235 auto buffer = nt_impl->get_unsafe_storage_as_tensor(); in alias_nested()
|
| D | NestedTensorBinaryOps.cpp | 87 f(self, other_impl->get_unsafe_storage_as_tensor()), in NestedTensor_elementwise_Tensor()
97 f(self_impl->get_unsafe_storage_as_tensor(), other), in NestedTensor_elementwise_Tensor()
175 f(self_impl->get_unsafe_storage_as_tensor(), in NestedTensor_elementwise_Tensor()
176 other_impl->get_unsafe_storage_as_tensor()), in NestedTensor_elementwise_Tensor()
|
| D | NestedTensorUnaryOps.cpp | 42 const Tensor& other_buffer = other_ptr->get_unsafe_storage_as_tensor(); in NestedTensor_where()
172 const auto& input_buffer = nt_input->get_unsafe_storage_as_tensor(); in _pin_memory_nested()
|
| D | NestedTensorMath.cpp | 432 const at::Tensor& buffer = self_ptr->get_unsafe_storage_as_tensor(); in select_nested()
485 const Tensor& input_buffer = input_ptr-> get_unsafe_storage_as_tensor(), in native_dropout_nested()
521 const Tensor& buffer = input_ptr->get_unsafe_storage_as_tensor(), in softmax_nested()
555 const Tensor& buffer = input_ptr->get_unsafe_storage_as_tensor(), in NestedTensor_all()
860 return nt_self->get_unsafe_storage_as_tensor(); in values_nested()
|
| D | NestedTensorMatmul.cpp | 31 …elf_buffer = self.is_nested() ? get_nested_tensor_impl(self)->get_unsafe_storage_as_tensor() : sel… in bmm_nested()
32 …at2_buffer = mat2.is_nested() ? get_nested_tensor_impl(mat2)->get_unsafe_storage_as_tensor() : mat… in bmm_nested()
|
| /external/pytorch/aten/src/ATen/ |
| D | NestedTensorImpl.h | 87 return get_unsafe_storage_as_tensor(); in get_buffer()
97 at::Tensor get_unsafe_storage_as_tensor() const { in get_unsafe_storage_as_tensor() function
|
| /external/pytorch/aten/src/ATen/native/nested/cuda/ |
| D | NestedTensorMatmul.cu | 331 …elf_buffer = self.is_nested() ? get_nested_tensor_impl(self)->get_unsafe_storage_as_tensor() : sel… in bmm_nested_cuda()
332 …at2_buffer = mat2.is_nested() ? get_nested_tensor_impl(mat2)->get_unsafe_storage_as_tensor() : mat… in bmm_nested_cuda()
|
| D | NestedTensorTransformerUtils.cpp | 178 Tensor storage_as_tensor = tensor_impl->get_unsafe_storage_as_tensor(); in view_as_dense()
|