Searched refs:element_shape_except_first_dim (Results 1 – 1 of 1) sorted by relevance
396 PartialTensorShape element_shape_except_first_dim; in Compute() local398 element_shape_except_first_dim = PartialTensorShape( in Compute()426 element_shape_except_first_dim = element_shape; in Compute()427 element_shape_except_first_dim.RemoveDim(0); in Compute()433 element_shape_except_first_dim.IsFullyDefined(), in Compute()437 element_shape_except_first_dim.DebugString())); in Compute()452 PartialTensorShape tmp = element_shape_except_first_dim; in Compute()460 &element_shape_except_first_dim)); in Compute()480 OP_REQUIRES(c, element_shape_except_first_dim.AsTensorShape(&output_shape), in Compute()484 element_shape_except_first_dim.DebugString())); in Compute()