Searched defs:cpp_tensor_stmts (Results 1 – 1 of 1) sorted by relevance
179 def set_cpp_tensors_requires_grad(cpp_tensor_stmts, python_tensors): argument189 def move_cpp_tensors_to_device(cpp_tensor_stmts, device): argument