Searched defs:input_unbind (Results 1 – 2 of 2) sorted by relevance
530 std::vector<Tensor> input_unbind = input.unbind(), in softmax_nested() local570 std::vector<Tensor> input_unbind = input.unbind(), in NestedTensor_all() local
423 std::vector<Tensor> input_unbind = at::unbind(input_sorted, 0); in _unique_dim_cpu_template() local