Searched refs:use_batch (Results 1 – 7 of 7) sorted by relevance
147 use_batch): argument167 if use_batch:207 use_batch=False): argument213 use_batch)215 use_batch)
288 PyObject* TFE_Py_ForwardAccumulatorNew(bool use_batch);
265 use_batch=True)282 use_batch=True)297 use_batch=True)
2450 bool use_batch) { argument2461 PyObject* to_batch = (use_batch) ? Py_True : Py_False;2588 bool use_batch = false) { argument2831 PyObject* TFE_Py_ForwardAccumulatorNew(bool use_batch) { argument2842 accumulator->accumulator = new ForwardAccumulator(*py_vspace, use_batch);3207 std::vector<PyObject*>* output_tangents, bool use_batch) { argument3209 output_tangents, use_batch);
69 self.use_batch = suite_common.ENABLE_CONTROLS_IN_BATCH105 suite_common.ENABLE_CONTROLS_IN_BATCH = self.use_batch180 self.use_batch = suite_common.ENABLE_CONTROLS_IN_BATCH197 suite_common.ENABLE_CONTROLS_IN_BATCH = self.use_batch
234 bool use_batch) in ForwardAccumulator() argument235 : vspace_(vspace), use_batch_(use_batch) { in ForwardAccumulator()
975 m.def("TFE_Py_ForwardAccumulatorNew", [](bool use_batch) { in PYBIND11_MODULE() argument976 return tensorflow::PyoOrThrow(TFE_Py_ForwardAccumulatorNew(use_batch)); in PYBIND11_MODULE()