Searched refs:partial_element_shape (Results 1 – 1 of 1) sorted by relevance
83 PartialTensorShape partial_element_shape; in Compute() local85 &partial_element_shape)); in Compute()88 partial_element_shape.IsFullyDefined() || in Compute()92 partial_element_shape.DebugString())); in Compute()100 PartialTensorShape tmp = partial_element_shape; in Compute()101 OP_REQUIRES_OK(c, tmp.MergeWith(t.shape(), &partial_element_shape)); in Compute()109 OP_REQUIRES(c, partial_element_shape.AsTensorShape(&element_shape), in Compute()113 partial_element_shape.DebugString())); in Compute()184 PartialTensorShape partial_element_shape; in Compute() local186 c, GetElementShapeFromInput(c, *l, 2, &partial_element_shape)); in Compute()[all …]