Searched refs:indices_inner_subshape (Results 1 – 1 of 1) sorted by relevance
1236 ShapeHandle indices_inner_subshape; in __anondb9326b21802() local1238 c->Subshape(indices_shape, batch_dims, &indices_inner_subshape)); in __anondb9326b21802()1242 c->Concatenate(params_outer_subshape, indices_inner_subshape, &out)); in __anondb9326b21802()