Searched refs:tail_it (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unroll_batch_matmul.cc | 35 Model* model, std::vector<std::unique_ptr<Operator>>::iterator* tail_it, in UnrollBatchMatMul3D() argument
60 *tail_it = model->operators.emplace(*tail_it, slice_a_op) + 1; in UnrollBatchMatMul3D()
73 *tail_it = model->operators.emplace(*tail_it, slice_a_reshape_op) + 1; in UnrollBatchMatMul3D()
92 *tail_it = model->operators.emplace(*tail_it, slice_b_op) + 1; in UnrollBatchMatMul3D()
105 *tail_it = model->operators.emplace(*tail_it, slice_b_reshape_op) + 1; in UnrollBatchMatMul3D()
114 *tail_it = model->operators.emplace(*tail_it, matmul_op) + 1; in UnrollBatchMatMul3D()
123 std::vector<std::unique_ptr<Operator>>::iterator* tail_it, in UnrollBatchMatMulRecursion() argument
136 model, tail_it, &batch_pack_inputs); in UnrollBatchMatMulRecursion()
144 input_lhs, input_rhs, batch_op, model, tail_it, new_batch_prefix); in UnrollBatchMatMulRecursion()
156 *tail_it = model->operators.emplace(*tail_it, pack_op) + 1; in UnrollBatchMatMulRecursion()
[all …]
|