Home
last modified time | relevance | path

Searched refs:batch_op_it (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dunroll_batch_matmul.cc217 auto batch_op_it = model->operators.begin() + op_index; in Run() local
218 if (batch_op_it->get()->type != OperatorType::kBatchMatMul) { in Run()
222 static_cast<const BatchMatMulOperator*>(batch_op_it->get()); in Run()
224 auto& tail_it = batch_op_it; in Run()
293 batch_op_it = model->operators.begin(); in Run()
294 for (; batch_op_it != model->operators.end(); ++batch_op_it) { in Run()
295 if (batch_op_it->get() == batch_op) { in Run()
299 CHECK(batch_op_it != model->operators.end()); in Run()
300 CHECK(batch_op_it->get() == batch_op); in Run()
301 model->operators.erase(batch_op_it); in Run()