Searched defs:first_dim (Results 1 – 5 of 5) sorted by relevance
25 const Index first_dim = Eigen::internal::array_get<0>(tensor.dimensions()); in run() local
140 Status CalculateOutputSize(INDEX_TYPE first_dim, OpKernelContext* c, in CalculateOutputSize()
307 int64 first_dim = -1; in Compute() local
275 const int64 first_dim = first_label - config.begin(); in EinsumDiagonalMask() local
1734 def _unflatten_first_dim(x, first_dim): argument3607 def _stack_tensor_list_shape(shape, first_dim): argument