Searched defs:use_batch (Results 1 – 4 of 4) sorted by relevance
147 use_batch): argument207 use_batch=False): argument
2450 bool use_batch) {2588 bool use_batch = false) {2831 PyObject* TFE_Py_ForwardAccumulatorNew(bool use_batch) {3207 std::vector<PyObject*>* output_tangents, bool use_batch) {
234 bool use_batch) in ForwardAccumulator()
975 m.def("TFE_Py_ForwardAccumulatorNew", [](bool use_batch) { in PYBIND11_MODULE()