Searched defs:self_buffer (Results 1 – 7 of 7) sorted by relevance
126 const Tensor& self_buffer = nt_self->get_buffer(); in _nested_sum_backward_cpu() local169 const Tensor& self_buffer = nt_self->get_buffer(); in _nested_select_backward_symint() local
151 const Tensor& self_buffer = self_ptr->get_unsafe_storage_as_tensor(), in clone_nested() local
31 …const Tensor& self_buffer = self.is_nested() ? get_nested_tensor_impl(self)->get_unsafe_storage_as… in bmm_nested() local
78 const Tensor& self_buffer = nt_self->get_buffer(); in NestedTensor_matmul() local
130 const auto self_buffer = self_ptr->get_buffer(); in NestedTensor_elementwise_Tensor() local
79 const auto self_buffer = self_ptr->get_buffer(); in _nested_op_dense_esuhm_kernel() local
331 …const Tensor &self_buffer = self.is_nested() ? get_nested_tensor_impl(self)->get_unsafe_storage_as… in bmm_nested_cuda() local