Searched defs:use_batch (Results 1 – 5 of 5) sorted by relevance
84 vTensor pack_weights(const Tensor& weight_arg, const bool use_batch = false) { in pack_weights()153 const bool use_batch = false) { in pack_biases()170 const bool use_batch = false) { in pack_biases_quantized_weights()349 const bool use_batch = false) { in available()400 const bool use_batch = false) { in usable()945 const bool use_batch) in LinearPackedContext()
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()