Searched refs:partial_element_shape (Results 1 – 1 of 1) sorted by relevance
188 PartialTensorShape partial_element_shape; in Compute() local190 &partial_element_shape)); in Compute()193 partial_element_shape.IsFullyDefined() || in Compute()197 partial_element_shape.DebugString())); in Compute()205 PartialTensorShape tmp = partial_element_shape; in Compute()206 OP_REQUIRES_OK(c, tmp.MergeWith(t.shape(), &partial_element_shape)); in Compute()214 OP_REQUIRES(c, partial_element_shape.AsTensorShape(&element_shape), in Compute()218 partial_element_shape.DebugString())); in Compute()292 PartialTensorShape partial_element_shape; in Compute() local294 c, GetElementShapeFromInput(c, *l, 2, &partial_element_shape)); in Compute()[all …]