Searched refs:element_shape_except_first_dim (Results 1 – 1 of 1) sorted by relevance
290 PartialTensorShape element_shape_except_first_dim; in Compute() local292 element_shape_except_first_dim = PartialTensorShape( in Compute()320 element_shape_except_first_dim = element_shape; in Compute()321 element_shape_except_first_dim.RemoveDim(0); in Compute()327 element_shape_except_first_dim.IsFullyDefined(), in Compute()331 element_shape_except_first_dim.DebugString())); in Compute()346 PartialTensorShape tmp = element_shape_except_first_dim; in Compute()354 &element_shape_except_first_dim)); in Compute()374 OP_REQUIRES(c, element_shape_except_first_dim.AsTensorShape(&output_shape), in Compute()378 element_shape_except_first_dim.DebugString())); in Compute()