Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dunroll_batch_matmul.cc127 const int current_dim_size = dims_vec[batch_prefix.size()]; in UnrollBatchMatMulRecursion() local
132 for (int batch = 0; batch < current_dim_size; ++batch) { in UnrollBatchMatMulRecursion()
140 for (int batch = 0; batch < current_dim_size; ++batch) { in UnrollBatchMatMulRecursion()