Searched defs:output_unbind (Results 1 – 6 of 6) sorted by relevance
54 std::vector<Tensor> output_unbind = output.unbind(); in NestedTensor_where() local
156 output_unbind = output.unbind(); in clone_nested() local
531 output_unbind = output.unbind(); in softmax_nested() local571 output_unbind = output.unbind(); in NestedTensor_all() local
63 std::vector<Tensor> output_unbind = output.unbind(); in bmm_nested() local
103 grad_unbind{grad.unbind()}, output_unbind{output.unbind()}; in nested_softmax_backward() local
397 std::vector<Tensor> output_unbind = output.unbind(); in bmm_nested_cuda() local