Searched defs:nt_sizes (Results 1 – 3 of 3) sorted by relevance
156 Tensor nt_sizes = nt_input->get_nested_sizes(); in NestedTensor_to_padded_tensor_cuda() local167 Tensor nt_sizes = nt_input->get_nested_sizes(); in NestedTensor_to_padded_tensor_cuda() local
381 Tensor nt_buffer, nt_sizes; in wrap_tensor_node() local
1000 const Tensor& nt_sizes, in unbind_backward_nested()2140 const Tensor& nt_sizes, in _nested_split_with_sizes_backward()