Searched refs:tensor_shape_except_first_dim (Results 1 – 1 of 1) sorted by relevance
383 shape_inference::ShapeHandle tensor_shape_except_first_dim; in __anon5e0c1ea50a02() local385 c->Subshape(tensor_shape, 1, &tensor_shape_except_first_dim)); in __anon5e0c1ea50a02()389 TF_RETURN_IF_ERROR(c->Merge(tensor_shape_except_first_dim, element_shape, in __anon5e0c1ea50a02()390 &tensor_shape_except_first_dim)); in __anon5e0c1ea50a02()