Searched defs:py_shape (Results 1 – 2 of 2) sorted by relevance
395 PyObject* py_shape = PyDict_GetItemString(cuda_dict, "shape"); in tensor_from_cuda_array_interface() local
1466 PyObject* py_shape = PyTuple_New(shape.dims()); in GetShape() local1482 PyObject* py_shape = GetShape(); in OnesLike() local1501 PyObject* py_shape = GetShape(); in ZerosLike() local