Searched refs:new_batch_prefix (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unroll_batch_matmul.cc | 133 std::vector<int> new_batch_prefix = batch_prefix; in UnrollBatchMatMulRecursion() local
134 new_batch_prefix.emplace_back(batch); in UnrollBatchMatMulRecursion()
135 UnrollBatchMatMul3D(input_lhs, input_rhs, batch_op, new_batch_prefix, in UnrollBatchMatMulRecursion()
141 std::vector<int> new_batch_prefix = batch_prefix; in UnrollBatchMatMulRecursion() local
142 new_batch_prefix.emplace_back(batch); in UnrollBatchMatMulRecursion()
144 input_lhs, input_rhs, batch_op, model, tail_it, new_batch_prefix); in UnrollBatchMatMulRecursion()
149 batch_op->outputs[0], "_b", absl::StrJoin(new_batch_prefix, "-")); in UnrollBatchMatMulRecursion()
|