Searched refs:py_vspace (Results 1 – 1 of 1) sorted by relevance
975 explicit PyVSpace(PyObject* py_vspace) : py_vspace_(py_vspace) { in PyVSpace() argument1146 PyVSpace* py_vspace = nullptr; variable1149 if (py_vspace != nullptr) { in TFE_Py_RegisterVSpace()1150 delete py_vspace; in TFE_Py_RegisterVSpace()1153 py_vspace = new PyVSpace(e); in TFE_Py_RegisterVSpace()1154 auto status = py_vspace->Initialize(); in TFE_Py_RegisterVSpace()1156 delete py_vspace; in TFE_Py_RegisterVSpace()1174 return py_vspace->GraphShape(absl::get<1>(shape_)); in GetShape()1735 *py_vspace, target_vec, sources_vec, source_tensors_that_are_targets,1763 result[i] = py_vspace->Zeros(tensor);