Searched defs:use_batch (Results 1 – 4 of 4) sorted by relevance
143 use_batch): argument203 use_batch=False): argument
2522 bool use_batch) {2660 bool use_batch = false) {2908 PyObject* TFE_Py_ForwardAccumulatorNew(bool use_batch) {3286 std::vector<PyObject*>* output_tangents, bool use_batch) {
234 bool use_batch) in ForwardAccumulator()
1134 m.def("TFE_Py_ForwardAccumulatorNew", [](bool use_batch) { in PYBIND11_MODULE()