Searched defs:first_dim (Results 1 – 3 of 3) sorted by relevance
25 const Index first_dim = Eigen::internal::array_get<0>(tensor.dimensions()); in run() local
331 int64 first_dim = -1; in Compute() local
1390 def _unflatten_first_dim(x, first_dim): argument