Searched refs:shapes_to_concat (Results 1 – 1 of 1) sorted by relevance
376 std::vector<TensorShape> shapes_to_concat; in Compute() local446 shapes_to_concat.push_back(std::move(expanded_tensor_shape)); in Compute()451 if (rank < 0) rank = shapes_to_concat[i].dims(); in Compute()452 OP_REQUIRES(context, rank == shapes_to_concat[i].dims(), in Compute()457 "] is: ", shapes_to_concat[i].dims())); in Compute()464 TensorShape preconcat_shape(shapes_to_concat[0]); in Compute()468 shapes_to_concat[i].dim_size(d))); in Compute()