Searched refs:input_unbind (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | Unique.cpp | 423 std::vector<Tensor> input_unbind = at::unbind(input_sorted, 0); in _unique_dim_cpu_template() local
425 input_unbind.begin(), input_unbind.end(), indices, inverse_indices, counts); in _unique_dim_cpu_template()
426 input_unbind.erase(last, input_unbind.end()); in _unique_dim_cpu_template()
427 counts = at::narrow(counts, 0, 0, input_unbind.size()); in _unique_dim_cpu_template()
430 auto output = at::stack(input_unbind, 0); in _unique_dim_cpu_template()
|
| /external/pytorch/aten/src/ATen/native/nested/ |
| D | NestedTensorMath.cpp | 530 std::vector<Tensor> input_unbind = input.unbind(), in softmax_nested() local
535 input_unbind[i], in softmax_nested()
570 std::vector<Tensor> input_unbind = input.unbind(), in NestedTensor_all() local
575 input_unbind[i], in NestedTensor_all()
|