Searched refs:existing_dim (Results 1 – 2 of 2) sorted by relevance
191 auto existing_dim = existing_dims[i]; in Compile() local196 OP_REQUIRES(ctx, existing_dim == 1, in Compile()199 " but dimension was not 1: ", existing_dim)); in Compile()202 new_shape.push_back(existing_dim); in Compile()206 if (existing_dim != 1) { in Compile()207 new_shape.push_back(existing_dim); in Compile()
222 auto existing_dim = existing_dims[i]; in Compute() local227 OP_REQUIRES(ctx, existing_dim == 1, in Compute()230 "], expected a dimension of 1, got ", existing_dim)); in Compute()233 new_shape.push_back(existing_dim); in Compute()237 if (existing_dim != 1) { in Compute()238 new_shape.push_back(existing_dim); in Compute()