Searched refs:py_vspace (Results 1 – 1 of 1) sorted by relevance
1193 explicit PyVSpace(PyObject* py_vspace) : py_vspace_(py_vspace) { in PyVSpace() argument1396 PyVSpace* py_vspace = nullptr; variable1401 if (py_vspace != nullptr) { in TFE_Py_RegisterVSpace()1411 delete py_vspace; in TFE_Py_RegisterVSpace()1414 py_vspace = new PyVSpace(e); in TFE_Py_RegisterVSpace()1415 auto status = py_vspace->Initialize(); in TFE_Py_RegisterVSpace()1417 delete py_vspace; in TFE_Py_RegisterVSpace()1435 return py_vspace->GraphShape(absl::get<1>(shape_)); in GetShape()1441 return py_vspace->OnesLike(tensor); in OnesLike()1445 PyObject* result = py_vspace->Ones(py_shape, dtype_field); in OnesLike()[all …]