experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

resolve name conflict for now

Browse Source 
PiperOrigin-RevId: 328437826
Change-Id: Iad3223ec7202d2ca8d9377bdf1a751288732eb6b
This commit is contained in:
Yanhua Sun 2020-08-25 17:34:52 -07:00 committed by TensorFlower Gardener
parent 29fd07dc0b
commit 9942d2f9e5
1 changed files with 7 additions and 8 deletions

15
tensorflow/python/eager/function.cc
View File

@ -26,17 +26,16 @@ namespace tensorflow {
namespace py = pybind11;
struct ConcreteFunction {
ConcreteFunction(){};
struct PyConcreteFunction {
PyConcreteFunction() {}
py::object _build_call_outputs(py::object result,
py::object structured_outputs,
bool _ndarrays_list, bool _ndarray_singleton);
};
py::object ConcreteFunction::_build_call_outputs(py::object result,
py::object structured_outputs,
bool _ndarrays_list,
bool _ndarray_singleton) {
py::object PyConcreteFunction::_build_call_outputs(
py::object result, py::object structured_outputs, bool _ndarrays_list,
bool _ndarray_singleton) {
static const py::module* nest =
new py::module(py::module::import("tensorflow.python.util.nest"));
// TODO(jlchu): Look into lazy loading of np_arrays module
@ -76,9 +75,9 @@ py::object ConcreteFunction::_build_call_outputs(py::object result,
}
PYBIND11_MODULE(_concrete_function, m) {
py::class_<ConcreteFunction>(m, "ConcreteFunction")
py::class_<PyConcreteFunction>(m, "ConcreteFunction")
.def(py::init<>())
.def("_build_call_outputs", &ConcreteFunction::_build_call_outputs);
.def("_build_call_outputs", &PyConcreteFunction::_build_call_outputs);
}
} // namespace tensorflow