Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dop_level_cost_estimator.cc1298 int s_dim = 1; in CountBatchMatMulOperations() local
1300 s_dim = smaller_rank_shape->dim(s_i).size(); in CountBatchMatMulOperations()
1303 batch_mat_mul->batch_dims.push_back(s_dim); in CountBatchMatMulOperations()
1305 num_matmuls *= std::max(b_dim, s_dim); in CountBatchMatMulOperations()